Home
last modified time | relevance | path

Searched defs:device_shape (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/tpu/
Dtpu_executable.cc112 Shape device_shape = ApiConverter::FromC(&c_device_shape); in HostShapeToDeviceShape() local
Dtpu_executable_interface.cc63 Shape device_shape = HostShapeToDeviceShape(host_shape); in AllocateOutputMemoryWithInputReuse() local
Dtpu_transfer_manager.cc61 xla::Shape device_shape = ApiConverter::FromC(&c_device_shape); in HostShapeToDeviceShape() local
/external/tensorflow/tensorflow/compiler/xrt/
Dxrt_state.cc416 xla::Shape* device_shape) { in ExpandTreeOfTuples()
463 xla::Shape device_shape; in MakeTuple() local
/external/tensorflow/tensorflow/core/tpu/
Dtpu_execute.cc112 xla::Shape device_shape = ApiConverter::FromC(&c_device_shape); in HostShapeToDeviceShape() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dtransfer_manager.cc204 Shape* device_shape) { in ReadDynamicShapes()
/external/tensorflow/tensorflow/compiler/jit/
Dxla_launch_util.cc236 const xla::Shape& device_shape = in PopulateInputs() local
/external/tensorflow/tensorflow/core/tpu/kernels/
Dtpu_execute_op.cc459 const xla::Shape& device_shape = in AllocateOutputTensors() local