Home
last modified time | relevance | path

Searched refs:device_sub_shape (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dshaped_buffer.cc72 TF_ASSIGN_OR_RETURN(const Shape* device_sub_shape, in SubShapedBuffer()
74 ShapedBuffer sub_shaped_buffer(*device_sub_shape, device_ordinal_); in SubShapedBuffer()
Dtransfer_manager.cc218 Shape& device_sub_shape = in ReadDynamicShapes() local
220 if (device_sub_shape.is_static()) { in ReadDynamicShapes()
244 device_sub_shape.mutable_dimensions()[i] = metadata.Get<int32>({i}); in ReadDynamicShapes()
/external/tensorflow/tensorflow/compiler/xrt/
Dxrt_state.cc369 const xla::Shape* device_sub_shape, in MakeSubBuffer()
374 *host_sub_shape, *device_sub_shape); in MakeSubBuffer()