| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_sharding.cc | 42 if (IsTuple()) { in ToString()
66 if (IsTuple()) { in UsesDevice()
77 CHECK(!ShapeUtil::IsTuple(tile_shape_)); in TileIndexForDevice()
79 CHECK(!IsTuple()); in TileIndexForDevice()
93 CHECK(!IsTuple()); in DeviceForTileIndex()
102 CHECK(!IsTuple()); in TileOffsetForDevice()
117 CHECK(!IsTuple()); in TileLimitForDevice()
128 if (IsTuple()) { in UniqueDevice()
148 if (!replicated_ && maximal_ && !IsTuple()) { in UniqueDevice()
156 if (IsTuple()) { in HasUniqueDevice()
[all …]
|
| D | hlo_sharding.h | 87 CHECK(ShapeUtil::IsTuple(tuple_shape)); in Tuple()
104 bool IsTuple() const { return tuple_; } in IsTuple() function
108 if (!IsTuple()) { in IsReplicated()
117 if (!IsTuple()) { in IsTileMaximal()
160 if (IsTuple()) { in GetAsShapeTree()
|
| D | tuple_util.cc | 24 CHECK(ShapeUtil::IsTuple(input_tuple->shape())); in ExtractPrefix()
44 CHECK(ShapeUtil::IsTuple(input_tuple->shape())); in AppendSuffix()
|
| D | hlo_element_type_converter.cc | 71 CHECK(!ShapeUtil::IsTuple(subshape)); in GetConvertedTupleShape()
95 CHECK(!ShapeUtil::IsTuple(ele_shape)); in ConvertTupleElements()
184 } else if (ShapeUtil::IsTuple(hlo->shape())) { in Run()
|
| D | hlo_value.cc | 48 ShapeUtil::IsTuple(instruction->shape()) ? (" " + index.ToString()) : ""; in ToString()
59 ShapeUtil::IsTuple(instruction->operand(operand_number)->shape()) in ToString()
90 string index_str = ShapeUtil::IsTuple(defining_instruction()->shape()) in ToShortString()
|
| D | allocation_tracker.cc | 103 if (!ShapeUtil::IsTuple(shaped_buffer->on_host_shape())) { in DeconstructTuple()
109 TF_RET_CHECK(ShapeUtil::IsTuple(shaped_buffer->on_device_shape())); in DeconstructTuple()
|
| D | zero_sized_hlo_elimination.cc | 35 ShapeUtil::IsTuple(instruction->shape())) { in Run()
|
| D | logical_buffer.cc | 34 is_tuple_ = ShapeUtil::IsTuple(s); in LogicalBuffer()
|
| D | logical_buffer.h | 139 bool IsTuple() const { return is_tuple_; } in IsTuple() function
|
| D | buffer_liveness_test.cc | 53 EXPECT_FALSE(ShapeUtil::IsTuple(a->shape())); in InstructionsMayInterfere()
54 EXPECT_FALSE(ShapeUtil::IsTuple(b->shape())); in InstructionsMayInterfere()
67 EXPECT_TRUE(ShapeUtil::IsTuple(a->shape())); in TupleElementsMayInterfere()
68 EXPECT_TRUE(ShapeUtil::IsTuple(b->shape())); in TupleElementsMayInterfere()
|
| D | shaped_buffer.cc | 89 if (ShapeUtil::IsTuple(subshape)) { in ToString()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | layout_util.cc | 91 DCHECK(!ShapeUtil::IsTuple(shape)); in GetDefaultLayoutForShape()
112 if (ShapeUtil::IsTuple(*shape)) { in SetToDefaultLayout()
144 if (ShapeUtil::IsTuple(shape)) { in ValidateLayoutInShape()
170 if (ShapeUtil::IsTuple(shape)) { in ValidateLayoutForShape()
266 if (ShapeUtil::IsTuple(shape) || !HasLayout(shape) || in IsPadded()
312 if (ShapeUtil::IsTuple(shape)) { in HasLayout()
387 if (ShapeUtil::IsTuple(src) != ShapeUtil::IsTuple(*dst)) { in CopyLayoutInternal()
391 if (ShapeUtil::IsTuple(src)) { in CopyLayoutInternal()
426 if (ShapeUtil::IsTuple(lhs) != ShapeUtil::IsTuple(rhs)) { in LayoutsInShapesEqual()
429 if (ShapeUtil::IsTuple(lhs)) { in LayoutsInShapesEqual()
|
| D | shape_util.cc | 73 if (ShapeUtil::IsTuple(lhs) || ShapeUtil::IsTuple(rhs)) { in CompareShapes()
74 return ShapeUtil::IsTuple(lhs) && ShapeUtil::IsTuple(rhs) && in CompareShapes()
156 CHECK(!ShapeUtil::IsTuple(shape)) in Rank()
321 return IsTuple(shape) && std::any_of(shape.tuple_shapes().begin(), in IsNestedTuple()
322 shape.tuple_shapes().end(), IsTuple); in IsNestedTuple()
326 return IsTuple(shape) && TupleElementCount(shape) == 0; in IsEmptyTuple()
330 return IsTuple(shape) ? IsEmptyTuple(shape) : HasZeroElements(shape); in IsNil()
334 CHECK(IsTuple(shape)) << HumanString(shape); in TupleElementCount()
340 CHECK(IsTuple(shape)); in GetTupleElementShape()
349 CHECK(IsTuple(tuple)); in SliceTuple()
[all …]
|
| D | shape_layout.cc | 55 CHECK(!ShapeUtil::IsTuple(shape_)); in layout()
64 CHECK(!ShapeUtil::IsTuple(shape_)); in ResetLayout()
|
| D | shape_util.h | 277 return !IsTuple(shape) && !IsOpaque(shape) && Rank(shape) == 0; in IsScalar()
280 return !IsTuple(shape) && !IsOpaque(shape) && TrueRank(shape) == 0; in IsEffectiveScalar()
384 static bool IsTuple(const Shape& shape) { in IsTuple() function
397 return !IsTuple(shape) && !IsOpaque(shape); in IsArray()
|
| /external/libcxx/test/std/utilities/tuple/tuple.tuple/tuple.cnstr/ |
| D | PR22806_constrain_tuple_like_ctor.pass.cpp | 32 struct IsTuple : std::false_type {}; struct
35 struct IsTuple<Tuple, std::tuple<Args...>> : std::true_type {}; struct
44 template <class Tuple, class = typename std::enable_if<IsTuple<Tuple>::value>::type>
57 template <class Tuple, class = typename std::enable_if<IsTuple<Tuple>::value>::type>
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | external_constant_pool.cc | 31 CHECK(!ShapeUtil::IsTuple(literal.shape())); in Insert()
|
| D | parallel_loop_emitter.cc | 34 CHECK(!ShapeUtil::IsTuple(shape_)); in EmitIndexAndSetExitBasicBlock()
|
| D | cpu_transfer_manager.cc | 99 if (!ShapeUtil::IsTuple(shape)) { in TransferLiteralToInfeed()
179 if (!ShapeUtil::IsTuple(literal_shape)) { in TransferLiteralFromOutfeed()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | literal_test_util.cc | 44 if (ShapeUtil::IsTuple(expected) != ShapeUtil::IsTuple(actual)) { in EqualShapes()
49 if (ShapeUtil::IsTuple(expected)) { in EqualShapes()
588 if (ShapeUtil::IsTuple(shape)) { in RecursiveElementCount()
696 if (ShapeUtil::IsTuple(expected.shape())) { in Near()
|
| D | client_test.cc | 97 EXPECT_TRUE(ShapeUtil::IsTuple(result->shape())); in XLA_TEST_F()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | shape_util.cc | 29 if (xla::ShapeUtil::IsTuple(shape)) { in XLAShapeToTensorShape()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | infeed_thunk.cc | 42 if (ShapeUtil::IsTuple(hlo_instruction()->shape())) { in ExecuteOnStream()
|
| D | gpu_transfer_manager.cc | 55 if (!ShapeUtil::IsTuple(shape)) { in TransferLiteralToInfeed()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | while_op.cc | 201 OP_REQUIRES(ctx, xla::ShapeUtil::IsTuple(body_input_shape), in Compile()
206 OP_REQUIRES(ctx, xla::ShapeUtil::IsTuple(cond_input_shape), in Compile()
|