Searched defs:xla_tensor (Results 1 – 5 of 5) sorted by relevance
25 XlaTensor* xla_tensor = in FromTensor() local36 const XlaTensor* xla_tensor = FromTensor(&tensor); in DeviceMemoryFromTensor() local
233 XlaTensor* xla_tensor = XlaTensor::FromTensor(t); in PopulateInputs() local241 const XlaTensor* xla_tensor = XlaTensor::FromTensor(t); in PopulateInputs() local341 if (XlaTensor* xla_tensor = XlaTensor::FromTensor(output_tensor)) { in PopulateOutputs() local358 XlaTensor* xla_tensor = XlaTensor::FromTensor(output_tensor); in PopulateOutputs() local423 XlaTensor* xla_tensor = XlaTensor::FromTensor(&output_tensor); in PopulateOutputs() local
125 XlaTensor* xla_tensor = XlaTensor::FromTensor(device_tensor); in CopyCPUTensorToDevice() local231 XlaTensor* xla_tensor = XlaTensor::FromTensor(device_tensor); in CopyDeviceTensorToCPU() local
120 if (const XlaTensor* xla_tensor = XlaTensor::FromTensor(&device_tensor)) { in Compile() local
118 const tensorflow::XlaTensor* xla_tensor = in DefaultPaddedShapeFn() local