Home
last modified time | relevance | path

Searched refs:is_multi_device (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/data/
Dcaptured_function.cc712 bool is_multi_device; in Instantiate() local
713 TF_RETURN_IF_ERROR(IsMultiDevice(ctx, &is_multi_device)); in Instantiate()
716 is_multi_device, instantiated_captured_function); in Instantiate()
734 bool* is_multi_device) const { in IsMultiDevice()
736 *is_multi_device = false; in IsMultiDevice()
768 *is_multi_device = true; in IsMultiDevice()
781 *is_multi_device = true; in IsMultiDevice()
794 *is_multi_device = true; in IsMultiDevice()
801 *is_multi_device = false; in IsMultiDevice()
809 CapturedFunction* captured_func, bool is_multi_device, in Create() argument
[all …]
Dcaptured_function.h192 Status IsMultiDevice(IteratorContext* ctx, bool* is_multi_device) const;
218 CapturedFunction* captured_func, bool is_multi_device,
277 CapturedFunction* captured_func, bool is_multi_device);