Searched defs:xla_tensor (Results 1 – 6 of 6) sorted by relevance
25 XlaTensor* xla_tensor = in FromTensor() local32 const XlaTensor* xla_tensor = FromTensor(&tensor); in DeviceMemoryFromTensor() local
133 XlaTensor* xla_tensor = XlaTensor::FromTensor(device_tensor); in CopyCPUTensorToDevice() local239 XlaTensor* xla_tensor = XlaTensor::FromTensor(device_tensor); in CopyDeviceTensorToCPU() local
262 XlaTensor* xla_tensor = XlaTensor::FromTensor(t); in PopulateInputs() local276 XlaTensor* xla_tensor = XlaTensor::FromTensor(t); in PopulateInputs() local334 XlaTensor* xla_tensor = XlaTensor::FromTensor(output_tensor); in PopulateXlaTensor() local
74 const tensorflow::XlaTensor* xla_tensor = in TpuPaddedShapeFn() local
68 const tensorflow::XlaTensor* xla_tensor = in DefaultPaddedShapeFn() local
215 XlaTensor* xla_tensor = XlaTensor::FromTensor(&tensor); in BuildComputationInputs() local315 XlaTensor* xla_tensor = XlaTensor::FromTensor(&tensor); in BuildComputationInputs() local479 XlaTensor* xla_tensor = XlaTensor::FromTensor(output_tensor); in AllocateOutputTensors() local