Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for TFE_TensorHandleDataType (0.19 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) {
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Fri Feb 09 07:47:20 GMT 2024
    - 25.4K bytes
    - Viewed (1)
Back to top