Home
last modified time | relevance | path

Searched refs:FromTensor (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/jit/
Dxla_tensor.cc21 /*static*/ XlaTensor* XlaTensor::FromTensor(const Tensor* tensor) { in FromTensor() function in tensorflow::XlaTensor
36 const XlaTensor* xla_tensor = FromTensor(&tensor); in DeviceMemoryFromTensor()
Dxla_launch_util.cc233 XlaTensor* xla_tensor = XlaTensor::FromTensor(t); in PopulateInputs()
241 const XlaTensor* xla_tensor = XlaTensor::FromTensor(t); in PopulateInputs()
341 if (XlaTensor* xla_tensor = XlaTensor::FromTensor(output_tensor)) { in PopulateOutputs()
358 XlaTensor* xla_tensor = XlaTensor::FromTensor(output_tensor); in PopulateOutputs()
423 XlaTensor* xla_tensor = XlaTensor::FromTensor(&output_tensor); in PopulateOutputs()
Dxla_device_context.cc125 XlaTensor* xla_tensor = XlaTensor::FromTensor(device_tensor); in CopyCPUTensorToDevice()
231 XlaTensor* xla_tensor = XlaTensor::FromTensor(device_tensor); in CopyDeviceTensorToCPU()
Dxla_tensor.h40 static XlaTensor* FromTensor(const Tensor* tensor);
Dxla_compile_on_demand_op.cc120 if (const XlaTensor* xla_tensor = XlaTensor::FromTensor(&device_tensor)) { in Compile()
Dxla_device.cc119 tensorflow::XlaTensor::FromTensor(&tensor); in DefaultPaddedShapeFn()