Home
last modified time | relevance | path

Searched defs:buffer_tree (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_transfer_manager.cc57 ShapeTree<InfeedBuffer> buffer_tree(literal_shape); in TransferLiteralToInfeed() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dtransfer_manager.cc342 se::Stream* stream, const ShapeTree<MaybeOwningDeviceMemory>& buffer_tree) { in WriteRootTupleIndexTable()
Dhlo_runner.cc130 ShapeTree<MaybeOwningDeviceMemory> buffer_tree( in ExecutionInputsFromScopedShapedBuffers() local