Home
last modified time | relevance | path

Searched refs:has_shaped_buffer (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/jit/
Dxla_tensor.h58 bool has_shaped_buffer() const { return shaped_buffer_.has_value(); } in has_shaped_buffer() function
62 CHECK(has_shaped_buffer()); in shaped_buffer()
66 CHECK(has_shaped_buffer()); in shaped_buffer()
Dxla_tpu_device.cc81 if (!xla_tensor->has_shaped_buffer()) { in TpuPaddedShapeFn()
195 TF_RET_CHECK(xla_input != nullptr && xla_input->has_shaped_buffer()); in TpuDeviceToDeviceCopy()
197 TF_RET_CHECK(xla_output != nullptr && !xla_output->has_shaped_buffer()); in TpuDeviceToDeviceCopy()
Dxla_tensor.cc34 CHECK(xla_tensor->has_shaped_buffer()); in DeviceMemoryFromTensor()
Dxla_device_context.cc143 TF_RET_CHECK(!xla_tensor->has_shaped_buffer()); in CopyCPUTensorToDevice()
Dxla_launch_util.cc277 CHECK(xla_tensor && xla_tensor->has_shaped_buffer()); in PopulateInputs()