Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 4 of 4 for TFE_TensorHandleDataType (0.38 sec)

  1. tensorflow/c/eager/parallel_device/parallel_device_lib.cc

      }
    
      TF_DataType dtype = TFE_TensorHandleDataType(components[0].get());
      // Verify that the TensorHandle's shape and dtype match all of the component
      // shapes and dtypes.
      for (TensorHandlePtr& component : components) {
        TFE_TensorHandleGetStatus(component.get(), status);
        if (!status->status.ok()) {
          return nullptr;
        }
        if (TFE_TensorHandleDataType(component.get()) != dtype) {
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Mon Oct 21 04:14:14 UTC 2024
    - 25.9K bytes
    - Viewed (0)
  2. tensorflow/c/eager/c_api_distributed_test.cc

      TFE_TensorHandle* packed_handle =
          TFE_CreatePackedTensorHandle(ctx, handles.data(), &num_replicas, status);
      ASSERT_EQ(TF_GetCode(status), TF_OK) << TF_Message(status);
      EXPECT_EQ(TFE_TensorHandleDataType(packed_handle), TF_RESOURCE);
      EXPECT_EQ(TFE_TensorHandleNumDims(packed_handle, status), 0);
      EXPECT_EQ(TFE_TensorHandleNumElements(packed_handle, status), 1);
    
      const string composite_device_name =
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 23.4K bytes
    - Viewed (0)
  3. tensorflow/c/eager/parallel_device/parallel_device_test.cc

      if (TF_GetCode(status) != TF_OK) return nullptr;
      TFE_OpSetDevice(op.get(), device, status);
      if (TF_GetCode(status) != TF_OK) return nullptr;
      TFE_OpSetAttrType(op.get(), "T", TFE_TensorHandleDataType(input));
      TFE_OpSetAttrInt(op.get(), "group_size", group_size);
      TFE_OpSetAttrInt(op.get(), "group_key", 0);
      TFE_OpSetAttrInt(op.get(), "instance_key", 0);
      const std::string merge_op("Add");
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Tue Aug 06 23:56:17 UTC 2024
    - 29.4K bytes
    - Viewed (0)
  4. tensorflow/c/eager/c_api.cc

      tsl::profiler::TraceMe activity("TFE_DeleteTensorHandle",
                                      tsl::profiler::TraceMeLevel::kInfo);
      if (h) {
        tensorflow::unwrap(h)->Unref();
      }
    }
    
    TF_DataType TFE_TensorHandleDataType(TFE_TensorHandle* h) {
      return static_cast<TF_DataType>(tensorflow::unwrap(h)->DataType());
    }
    
    int TFE_TensorHandleNumDims(TFE_TensorHandle* h, TF_Status* status) {
      if (h == nullptr) {
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 43.9K bytes
    - Viewed (0)
Back to top