Home
last modified time | relevance | path

Searched defs:expected_device (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/runtime_fallback/kernel/gpu/
Dconversion_function.cc47 auto expected_device = GetTfDevice(exec_ctx, src); in ConvertKernelFallbackTensorOnGpuToDenseGpuTensor() local
82 auto expected_device = GetTfDevice(exec_ctx, src); in ConvertDenseGpuTensorToKernelFallbackTensor() local
/external/tensorflow/tensorflow/core/runtime_fallback/runtime/
Druntime_fallback_op_handler.cc182 auto expected_device = GetDeviceFromFallbackTensor( in GetResultDevice() local
203 auto expected_device = GetDeviceFromFallbackTensor( in GetResultDevice() local
/external/tensorflow/tensorflow/python/lib/core/
Dpy_func.cc152 const Device* expected_device, in ExtractTensorFromEagerTensor()
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dexecute.cc225 Device* expected_device = kernel->InputDevice(i); in ValidateInputTypeAndPlacement() local