Searched refs:XlaTensor (Results 1 – 7 of 7) sorted by relevance
21 /*static*/ XlaTensor* XlaTensor::FromTensor(const Tensor* tensor) { in FromTensor()25 XlaTensor* xla_tensor = in FromTensor()30 /*static*/ bool XlaTensor::RefCountIsOne(const Tensor& tensor) { in RefCountIsOne()34 /*static*/ se::DeviceMemoryBase XlaTensor::DeviceMemoryFromTensor( in DeviceMemoryFromTensor()36 const XlaTensor* xla_tensor = FromTensor(&tensor); in DeviceMemoryFromTensor()46 Status XlaTensor::AllocateShapedBuffer(DataType dtype, in AllocateShapedBuffer()75 void XlaTensor::WaitForDefinitionEventOnStream(se::Stream* stream) { in WaitForDefinitionEventOnStream()93 void XlaTensor::ResetDefinitionEvent(std::shared_ptr<se::Event> event, in ResetDefinitionEvent()107 /*static*/ XlaTensor* XlaTensor::FromOpaquePointer(void* ptr) { in FromOpaquePointer()110 return reinterpret_cast<XlaTensor*>(value & ~kTag); in FromOpaquePointer()[all …]
36 class XlaTensor {40 static XlaTensor* FromTensor(const Tensor* tensor);106 static XlaTensor* FromOpaquePointer(void* ptr);108 static void* ToOpaquePointer(XlaTensor* tensor);
48 return XlaTensor::ToOpaquePointer(new XlaTensor()); in AllocateRaw()52 delete XlaTensor::FromOpaquePointer(ptr); in DeallocateRaw()125 XlaTensor* xla_tensor = XlaTensor::FromTensor(device_tensor); in CopyCPUTensorToDevice()231 XlaTensor* xla_tensor = XlaTensor::FromTensor(device_tensor); in CopyDeviceTensorToCPU()
233 XlaTensor* xla_tensor = XlaTensor::FromTensor(t); in PopulateInputs()241 const XlaTensor* xla_tensor = XlaTensor::FromTensor(t); in PopulateInputs()250 se::DeviceMemoryBase dmem = XlaTensor::DeviceMemoryFromTensor(*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()
120 if (const XlaTensor* xla_tensor = XlaTensor::FromTensor(&device_tensor)) { in Compile()
118 const tensorflow::XlaTensor* xla_tensor = in DefaultPaddedShapeFn()119 tensorflow::XlaTensor::FromTensor(&tensor); in DefaultPaddedShapeFn()
616 friend class XlaTensor; // For access to RefCountIsOne(). variable