Searched refs:CanonicalDevice (Results 1 – 5 of 5) sorted by relevance
116 const Device* d = ctx_->CanonicalDevice(kernel_->InputDevice(i)); in Run()194 const Device* d = ctx_->CanonicalDevice(kernel_->InputDevice(i)); in Run()221 handle->Poison(status, ctx_->CanonicalDevice(kernel_->OutputDevice(i))); in Abort()
634 Device* output_device = ctx.CanonicalDevice(kernel->OutputDevice(i)); in AddOrExecuteNode()994 Device* output_device = ctx->CanonicalDevice(kernel->OutputDevice(i)); in GetKernelOutputs()1019 if (TF_PREDICT_FALSE(ctx->CanonicalDevice(kernel->OutputDevice(i)) != in GetKernelOutputs()1030 ctx->CanonicalDevice(kernel->OutputDevice(i)))); in GetKernelOutputs()1151 Device* d = ctx->CanonicalDevice(dstd); in LocalEagerCopyToDevice()1239 Device* d = ctx->CanonicalDevice(device); in EagerCopyToDevice()1406 /* d= */ ctx.CanonicalDevice(kernel->OutputDevice(i)), in EagerLocalExecuteAsync()
79 Status s = in->TensorValue(ctx->CanonicalDevice(d), &tensor_args_flat[i]); in Init()
201 Device* CanonicalDevice(Device* d) const { in CanonicalDevice() function
108 Device* d = ctx_->CanonicalDevice(absl::get<Device*>(op->Device())); in RunLocalSend()274 Device* recv_device = ctx_->CanonicalDevice(recv_device_); in StartRecv()