Searched defs:expected_device (Results 1 – 8 of 8) sorted by relevance
47 auto expected_device = GetTfDevice(exec_ctx, src); in ConvertKernelFallbackTensorOnGpuToDenseGpuTensor() local82 auto expected_device = GetTfDevice(exec_ctx, src); in ConvertDenseGpuTensorToKernelFallbackTensor() local
182 auto expected_device = GetDeviceFromFallbackTensor( in GetResultDevice() local203 auto expected_device = GetDeviceFromFallbackTensor( in GetResultDevice() local
87 auto expected_device = tensors[0].device(); in preprocess_chunk_cat_inputs() local
129 auto expected_device = scaled_grads[0].device(); in _amp_foreach_non_finite_check_and_unscale_cuda_() local
152 const Device* expected_device, in ExtractTensorFromEagerTensor()
61 self, fn, in_devices, expected_device, in_shapes=None, subtest_str="" argument
225 Device* expected_device = kernel->InputDevice(i); in ValidateInputTypeAndPlacement() local
922 cl_device_id expected_device = make_device_id(0); in testCommandQueueFromSpecifiedContext() local3860 cl_device_id expected_device = make_device_id(0); in testSemaphoreWithProperties() local