Searched refs:copied_tensor (Results 1 – 3 of 3) sorted by relevance
68 Tensor* copied_tensor; in Compute() local70 &copied_tensor)); in Compute()83 device, device_ctxt, &src_tensor, copied_tensor, in Compute()88 *copied_tensor = tensor::DeepCopy(src_tensor); in Compute()97 SYCLmemcpy(context->eigen_sycl_device(), src_tensor, copied_tensor); in Compute()99 *copied_tensor = tensor::DeepCopy(src_tensor); in Compute()102 *copied_tensor = tensor::DeepCopy(src_tensor); in Compute()
45 is_ref](const Tensor* copied_tensor) { in DebugGateway() argument46 val_cb_(node_name, output_slot, *copied_tensor, is_ref); in DebugGateway()
527 TFE_TensorHandle* copied_tensor = TFE_TensorHandleCopyToDevice( in ValidateInputTypeAndPlacement() local537 op->inputs[i] = copied_tensor->t; in ValidateInputTypeAndPlacement()538 copied_tensors->push_back(copied_tensor); in ValidateInputTypeAndPlacement()539 op->input_devices[i] = copied_tensor->d; in ValidateInputTypeAndPlacement()