Searched refs:IsMultiDevice (Results 1 – 5 of 5) sorted by relevance
192 Status IsMultiDevice(IteratorContext* ctx, bool* is_multi_device) const;
713 TF_RETURN_IF_ERROR(IsMultiDevice(ctx, &is_multi_device)); in Instantiate()733 Status CapturedFunction::IsMultiDevice(IteratorContext* ctx, in IsMultiDevice() function in tensorflow::data::CapturedFunction
335 ProcessFunctionLibraryRuntime::IsMultiDevice( in IsMultiDevice() function in tensorflow::ProcessFunctionLibraryRuntime1019 MultiDeviceFunctionData* data = IsMultiDevice(handle); in GetOutputDevices()1083 const MultiDeviceFunctionData* data = IsMultiDevice(handle); in RunMultiDevice()1334 if (IsMultiDevice(handle)) { in ReleaseHandle()1646 const MultiDeviceFunctionData* data = IsMultiDevice(handle); in Run()
305 MultiDeviceFunctionData* IsMultiDevice(
586 if (parent_->IsMultiDevice(h)) { in GetRetTypes()