Searched defs:xla_tensor (Results 1 – 8 of 8) sorted by relevance
117 XlaTensor* xla_tensor = XlaTensor::FromTensor(&tensor); in BuildInputBuffers() local182 XlaTensor* xla_tensor = XlaTensor::FromTensor(&tensor); in BuildInputBuffers() local293 XlaTensor* xla_tensor = XlaTensor::FromTensor(output_tensor); in UpdateOutputVariables() local
214 XlaTensor* xla_tensor = XlaTensor::FromTensor(&tensor); in BuildComputationInputs() local314 XlaTensor* xla_tensor = XlaTensor::FromTensor(&tensor); in BuildComputationInputs() local478 XlaTensor* xla_tensor = XlaTensor::FromTensor(output_tensor); in AllocateOutputTensors() local
27 XlaTensor* xla_tensor = in FromTensor() local34 const XlaTensor* xla_tensor = FromTensor(&tensor); in DeviceMemoryFromTensor() local
133 XlaTensor* xla_tensor = XlaTensor::FromTensor(device_tensor); in CopyCPUTensorToDevice() local241 XlaTensor* xla_tensor = XlaTensor::FromTensor(device_tensor); in CopyDeviceTensorToCPU() local
290 XlaTensor* xla_tensor = XlaTensor::FromTensor(t); in PopulateInputs() local360 XlaTensor* xla_tensor = XlaTensor::FromTensor(&output_tensor); in GetOrCreateTensorForOutput() local
77 const tensorflow::XlaTensor* xla_tensor = in TpuPaddedShapeFn() local
68 const tensorflow::XlaTensor* xla_tensor = in DefaultPaddedShapeFn() local
59 static inline void AllEqual(at::Tensor tensor, at::Tensor xla_tensor) { in AllEqual()