Home
last modified time | relevance | path

Searched refs:HasCustomDeviceInput (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/eager/
Dcustom_device_op_handler.cc68 if (op->HasCustomDeviceInput()) { in Execute()
135 !op.HasCustomDeviceInput()) { in MaybePinToCustomDevice()
Deager_operation.cc431 if (TF_PREDICT_TRUE(!HasCustomDeviceInput())) { in TensorHandleInputs()
442 if (TF_PREDICT_TRUE(!HasCustomDeviceInput())) { in MutableTensorHandleInputs()
Deager_operation.h90 bool HasCustomDeviceInput() const override { in HasCustomDeviceInput() function
/external/tensorflow/tensorflow/c/eager/
Dimmediate_execution_operation.h56 virtual bool HasCustomDeviceInput() const = 0;
/external/tensorflow/tensorflow/core/tfrt/eager/cpp_tests/core_runtime/
Dop_handler_selector_test.cc174 bool HasCustomDeviceInput() const override { return false; } in HasCustomDeviceInput() function in tfrt::tf::__anond52f36620111::FakeOperation
/external/tensorflow/tensorflow/core/tfrt/eager/
Dc_api_tfrt.h498 bool HasCustomDeviceInput() const override { in HasCustomDeviceInput() function