Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 31 for Type (0.12 sec)

  1. tensorflow/c/eager/c_api_experimental_test.cc

      TFE_TensorHandle* hcpu = TestMatrixTensorHandle(ctx);
      const char* device_type = TFE_TensorHandleDeviceType(hcpu, status.get());
      ASSERT_EQ(TF_OK, TF_GetCode(status.get())) << TF_Message(status.get());
      ASSERT_TRUE(absl::StrContains(device_type, "CPU")) << device_type;
      int device_id = TFE_TensorHandleDeviceID(hcpu, status.get());
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Aug 03 03:14:26 GMT 2023
    - 31.5K bytes
    - Viewed (1)
  2. tensorflow/c/experimental/filesystem/modular_filesystem.cc

      memset(&option, 0, sizeof(option));
      option.name = const_cast<char*>(name.c_str());
      TF_Filesystem_Option_Value option_value;
      memset(&option_value, 0, sizeof(option_value));
      option_value.type_tag = TF_Filesystem_Option_Type_Buffer;
      option_value.num_values = values.size();
      std::vector<TF_Filesystem_Option_Value_Union> option_values(values.size());
      for (size_t i = 0; i < values.size(); i++) {
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Wed Sep 06 19:12:29 GMT 2023
    - 23.1K bytes
    - Viewed (0)
  3. tensorflow/c/eager/c_api_unified_experimental_test.cc

      ASSERT_EQ(TF_OK, TF_GetCode(status.get())) << TF_Message(status.get());
    
      // Test that full type information can be accessed.
      auto outs = unwrap(add_outputs);
      auto h = outs->outputs[0];
      ASSERT_NE(h, nullptr);
      ASSERT_EQ(h->FullType().type_id(), TFT_UNSET);
      ASSERT_EQ(unwrap(inputs[0])->FullType().type_id(), TFT_UNSET);
    
      // Clean up operation and inputs.
      TF_DeleteAbstractOp(add_op);
    
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Fri May 19 21:44:52 GMT 2023
    - 39.1K bytes
    - Viewed (0)
  4. tensorflow/c/eager/dlpack.cc

      std::string device_type = parsed_name.type;
      int device_id = 0;
      if (parsed_name.has_id) {
        device_id = parsed_name.id;
      }
    
      ctx.device_id = device_id;
      if (device_type == "CPU") {
        ctx.device_type = DLDeviceType::kDLCPU;
      } else if (device_type == "GPU") {
    #if TENSORFLOW_USE_ROCM
        ctx.device_type = DLDeviceType::kDLROCM;
    #else
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Feb 15 09:49:45 GMT 2024
    - 12.8K bytes
    - Viewed (0)
  5. tensorflow/c/experimental/grappler/grappler.h

      void* ext;  // reserved for future use
    
      // Graph C API version.
      int32_t major_version;
      int32_t minor_version;
      int32_t patch_version;
    
      // Backend device type supported by the optimizer.
      const char* device_type;
      TP_OptimizerConfigs* optimizer_configs;  // output, set by plugin
      TP_Optimizer* optimizer;                 // output, set by plugin
    } TP_OptimizerRegistrationParams;
    
    C
    - Registered: Tue Feb 27 12:39:08 GMT 2024
    - Last Modified: Wed Aug 03 18:08:43 GMT 2022
    - 12.5K bytes
    - Viewed (0)
  6. tensorflow/c/experimental/grappler/grappler.cc

      TF_RETURN_IF_ERROR(ValidateTPOptimizerConfigs(optimizer_configs));
    
      CGraphOptimizerRegister(
          [=]() { return new CGraphOptimizer(optimizer, params.device_type); },
          optimizer_configs, params.device_type);
    
      return absl::OkStatus();
    }
    
    }  // namespace grappler
    }  // namespace tensorflow
    
    void TF_GetNodesToPreserveListSize(const TF_GrapplerItem* item, int* num_values,
    C++
    - Registered: Tue Feb 27 12:39:08 GMT 2024
    - Last Modified: Wed Sep 06 19:12:29 GMT 2023
    - 15K bytes
    - Viewed (1)
  7. tensorflow/c/eager/gradients.cc

          gtl::ArraySlice<AbstractTensorHandle*> gradient_tensors) const override;
    
      // Calls the passed-in backward function.
      // op_type is the op's name provided in RecordOperation.
      Status CallBackwardFunction(
          const string& op_type, GradientFunction* gradient_function,
          const std::vector<int64_t>& unneeded_gradients,
          gtl::ArraySlice<AbstractTensorHandle*> output_gradients,
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Feb 15 09:49:45 GMT 2024
    - 19.3K bytes
    - Viewed (0)
  8. tensorflow/c/eager/c_api_unified_experimental_graph.cc

        }
        if (raw_device_name) {
          device_name_ = raw_device_name;
        }
        op_type_ = op;
        return absl::OkStatus();
      }
      Status SetOpName(const char* const op_name) override {
        if (op_) {
          return errors::FailedPrecondition(
              "SetOpName called on already built op.");
        }
        if (op_type_.empty()) {
          return errors::FailedPrecondition(
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Tue Mar 12 20:00:09 GMT 2024
    - 15.4K bytes
    - Viewed (1)
  9. tensorflow/c/eager/c_api.cc

          }
          // Type
          if (const int type_size = default_value.list().type_size()) {
            absl::InlinedVector<unsigned int, 4> type_vector;
            type_vector.reserve(type_size);
            for (int i = 0; i < type_size; ++i) {
              type_vector.push_back(default_value.list().type(i));
            }
            TFE_OpSetAttrTypeList(
                op, attr_name,
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Tue Mar 12 20:00:09 GMT 2024
    - 43.9K bytes
    - Viewed (2)
  10. tensorflow/c/c_test_util.cc

        return false;
      }
      bool found_dtype = false;
      bool found_shape = false;
      for (const auto& attr : node_def.attr()) {
        if (attr.first == "dtype") {
          if (attr.second.type() == tensorflow::DT_INT32) {
            found_dtype = true;
          } else {
            return false;
          }
        } else if (attr.first == "shape") {
          found_shape = true;
        }
      }
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Fri Oct 15 03:16:52 GMT 2021
    - 17.8K bytes
    - Viewed (2)
Back to top