- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for TFE_TensorHandleDataType (0.38 sec)
-
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) -
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) -
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) -
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)