Home
last modified time | relevance | path

Searched defs:literal_shape (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/
Dpacked_literal_reader.cc46 Shape literal_shape = shape; in Read() local
Dliteral_test.cc1607 const Shape literal_shape = ShapeUtil::MakeShape(S64, {3}); in TEST_F() local
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_transfer_manager.cc148 se::StreamExecutor* /*executor*/, const Shape& literal_shape, in TransferLiteralFromOutfeed()
/external/tensorflow/tensorflow/compiler/xla/service/
Dgeneric_transfer_manager.cc160 se::StreamExecutor* executor, const Shape& literal_shape, in TransferLiteralFromOutfeed()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_transfer_manager.cc178 se::StreamExecutor* executor, const Shape& literal_shape, in TransferLiteralFromOutfeed()