Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dshaped_buffer.cc163 const xla::Shape& sub_on_device_shape = in TakeSubTree() local
166 ScopedShapedBuffer output(sub_on_host_shape, sub_on_device_shape, in TakeSubTree()