Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A 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
H A 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
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/eager/
H A Dexecute_node_test.cc67 std::unique_ptr<CompositeDevice> composite_device = in TEST() local
H A Dtensor_handle.cc323 CompositeDevice* composite_device = nullptr; in CreatePackedHandle() local
H A Dcontext.cc1153 CompositeDevice** composite_device) { in FindOrCreateCompositeDevice()
H A Dexecute.cc985 CompositeDevice* composite_device = nullptr; in ExtractFunctionInputInfo() local
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/cc/
H A Ddtensor_device_util.h185 CompositeDevice* composite_device() const { return composite_device_; } in composite_device() function
/aosp_15_r20/external/tensorflow/tensorflow/core/data/
H A Dcaptured_function.cc594 string composite_device; in Instantiate() local
/aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/eager/
H A Deager_service_impl_test.cc1131 const string composite_device = in TEST_F() local
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Dfunction.cc1042 string composite_device; in GetFunctionResourceInputDevice() local