Home
last modified time | relevance | path

Searched defs:composite_device (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/
Dcomposite_device_test.cc29 std::unique_ptr<CompositeDevice> composite_device = in TEST() local
45 std::unique_ptr<CompositeDevice> composite_device = in TEST() local
57 std::unique_ptr<CompositeDevice> composite_device = in TEST() local
75 std::unique_ptr<CompositeDevice> composite_device = in TEST() local
Dprocess_function_library_runtime_test.cc909 std::unique_ptr<CompositeDevice> composite_device = in TEST_F() local
1245 std::unique_ptr<CompositeDevice> composite_device = in TEST_F() local
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dexecute_node_test.cc67 std::unique_ptr<CompositeDevice> composite_device = in TEST() local
Dtensor_handle.cc323 CompositeDevice* composite_device = nullptr; in CreatePackedHandle() local
Dexecute.cc985 CompositeDevice* composite_device = nullptr; in ExtractFunctionInputInfo() local
Dcontext.cc1153 CompositeDevice** composite_device) { in FindOrCreateCompositeDevice()
/external/tensorflow/tensorflow/dtensor/cc/
Ddtensor_device_util.h185 CompositeDevice* composite_device() const { return composite_device_; } in composite_device() function
/external/tensorflow/tensorflow/core/data/
Dcaptured_function.cc594 string composite_device; in Instantiate() local
/external/tensorflow/tensorflow/core/distributed_runtime/eager/
Deager_service_impl_test.cc1131 const string composite_device = in TEST_F() local
/external/tensorflow/tensorflow/core/framework/
Dfunction.cc1042 string composite_device; in GetFunctionResourceInputDevice() local