Searched refs:literal_shape (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_transfer_manager.cc | 178 se::StreamExecutor* executor, const Shape& literal_shape, in TransferLiteralFromOutfeed() argument
180 if (!literal_shape.IsTuple()) { in TransferLiteralFromOutfeed()
181 int64 size = GetByteSizeRequirement(literal_shape); in TransferLiteralFromOutfeed()
185 absl::bit_cast<const int64*>(literal_shape.dimensions().data()), in TransferLiteralFromOutfeed()
186 literal_shape.dimensions().size()); in TransferLiteralFromOutfeed()
194 << ShapeUtil::HumanString(literal_shape); in TransferLiteralFromOutfeed()
200 if (ShapeUtil::IsNestedTuple(literal_shape)) { in TransferLiteralFromOutfeed()
206 for (int64 i = 0; i < literal_shape.tuple_shapes_size(); ++i) { in TransferLiteralFromOutfeed()
208 ShapeUtil::GetTupleElementShape(literal_shape, i); in TransferLiteralFromOutfeed()
216 TF_RET_CHECK(ShapeUtil::Compatible(received_shape, literal_shape)) in TransferLiteralFromOutfeed()
[all …]
|
| D | cpu_transfer_manager.h | 44 const Shape& literal_shape,
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | packed_literal_reader.cc | 46 Shape literal_shape = shape; in Read() local
49 LayoutUtil::ValidateLayoutForShape(*layout, literal_shape)); in Read()
50 *literal_shape.mutable_layout() = *layout; in Read()
59 Literal result(literal_shape); in Read()
|
| D | literal_test.cc | 1607 const Shape literal_shape = ShapeUtil::MakeShape(S64, {3}); in TEST_F() local
1610 literal_shape); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_transfer_manager.cc | 148 se::StreamExecutor* /*executor*/, const Shape& literal_shape, in TransferLiteralFromOutfeed() argument
151 &literal_shape); in TransferLiteralFromOutfeed()
157 const Shape& shape = ShapeUtil::GetSubshape(literal_shape, index); in TransferLiteralFromOutfeed()
|
| D | gpu_transfer_manager.h | 44 const Shape& literal_shape,
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | generic_transfer_manager.h | 56 const Shape& literal_shape,
|
| D | generic_transfer_manager.cc | 160 se::StreamExecutor* executor, const Shape& literal_shape, in TransferLiteralFromOutfeed() argument
|
| D | transfer_manager.h | 193 se::StreamExecutor* executor, const Shape& literal_shape,
|