| /external/tensorflow/tensorflow/compiler/xla/service/ |
| zero_sized_hlo_elimination.cc | 35 ShapeUtil::IsTuple(instruction->shape())) {
|
| tuple_util.cc | 24 CHECK(ShapeUtil::IsTuple(input_tuple->shape()));
44 CHECK(ShapeUtil::IsTuple(input_tuple->shape()));
|
| hlo_sharding.cc | 42 if (IsTuple()) {
66 if (IsTuple()) {
77 CHECK(!ShapeUtil::IsTuple(tile_shape_));
79 CHECK(!IsTuple());
93 CHECK(!IsTuple());
102 CHECK(!IsTuple());
117 CHECK(!IsTuple());
128 if (IsTuple()) {
148 if (!replicated_ && maximal_ && !IsTuple()) {
156 if (IsTuple()) {
[all...] |
| hlo_sharding.h | 87 CHECK(ShapeUtil::IsTuple(tuple_shape));
104 bool IsTuple() const { return tuple_; }
108 if (!IsTuple()) {
117 if (!IsTuple()) {
128 // REQUIRES: !IsTuple()
133 // REQUIRES: !IsTuple()
139 // REQUIRES: !IsTuple()
145 // REQUIRES: !IsTuple()
149 // REQUIRES: !IsTuple&& !Replicated() && IsTileMaximal()
156 // tuple, if IsTuple, or a ShapeTree with a single element containing thi
[all...] |
| logical_buffer.cc | 34 is_tuple_ = ShapeUtil::IsTuple(s);
|
| logical_buffer.h | 139 bool IsTuple() const { return is_tuple_; }
|
| hlo_element_type_converter.cc | 71 CHECK(!ShapeUtil::IsTuple(subshape));
95 CHECK(!ShapeUtil::IsTuple(ele_shape));
184 } else if (ShapeUtil::IsTuple(hlo->shape())) {
|
| hlo_value.cc | 48 ShapeUtil::IsTuple(instruction->shape()) ? (" " + index.ToString()) : "";
59 ShapeUtil::IsTuple(instruction->operand(operand_number)->shape())
90 string index_str = ShapeUtil::IsTuple(defining_instruction()->shape())
|
| allocation_tracker.cc | 103 if (!ShapeUtil::IsTuple(shaped_buffer->on_host_shape())) {
109 TF_RET_CHECK(ShapeUtil::IsTuple(shaped_buffer->on_device_shape()));
|
| generic_transfer_manager.cc | 86 if (!ShapeUtil::IsTuple(subshape)) {
|
| shaped_buffer.cc | 89 if (ShapeUtil::IsTuple(subshape)) {
|
| while_util.cc | 94 CHECK(ShapeUtil::IsTuple(while_instr->shape()));
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| shape_layout.cc | 55 CHECK(!ShapeUtil::IsTuple(shape_));
64 CHECK(!ShapeUtil::IsTuple(shape_));
|
| layout_util.cc | 91 DCHECK(!ShapeUtil::IsTuple(shape));
112 if (ShapeUtil::IsTuple(*shape)) {
144 if (ShapeUtil::IsTuple(shape)) {
170 if (ShapeUtil::IsTuple(shape)) {
266 if (ShapeUtil::IsTuple(shape) || !HasLayout(shape) ||
312 if (ShapeUtil::IsTuple(shape)) {
387 if (ShapeUtil::IsTuple(src) != ShapeUtil::IsTuple(*dst)) {
391 if (ShapeUtil::IsTuple(src)) {
426 if (ShapeUtil::IsTuple(lhs) != ShapeUtil::IsTuple(rhs))
[all...] |
| shape_util.cc | 73 if (ShapeUtil::IsTuple(lhs) || ShapeUtil::IsTuple(rhs)) {
74 return ShapeUtil::IsTuple(lhs) && ShapeUtil::IsTuple(rhs) &&
156 CHECK(!ShapeUtil::IsTuple(shape))
321 return IsTuple(shape) && std::any_of(shape.tuple_shapes().begin(),
322 shape.tuple_shapes().end(), IsTuple);
326 return IsTuple(shape) && TupleElementCount(shape) == 0;
330 return IsTuple(shape) ? IsEmptyTuple(shape) : HasZeroElements(shape);
334 CHECK(IsTuple(shape)) << HumanString(shape)
[all...] |
| shape_util.h | 154 // Precondition: !IsTuple(shape)
165 // Precondition: (!ShapeUtil::IsTuple(shape) || pointer_size > 0) &&
171 // Precondition: !ShapeUtil::IsOpaque(shape) && !ShapeUtil::IsTuple(shape)
261 // Precondition: !IsTuple(shape)
277 return !IsTuple(shape) && !IsOpaque(shape) && Rank(shape) == 0;
280 return !IsTuple(shape) && !IsOpaque(shape) && TrueRank(shape) == 0;
384 static bool IsTuple(const Shape& shape) {
397 return !IsTuple(shape) && !IsOpaque(shape);
411 // Precondition: IsTuple(shape)
415 // Precondition: IsTuple(shape) && TupleElementCount(shape) > inde
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| external_constant_pool.cc | 31 CHECK(!ShapeUtil::IsTuple(literal.shape()));
|
| parallel_loop_emitter.cc | 34 CHECK(!ShapeUtil::IsTuple(shape_));
|
| /external/libcxx/test/std/utilities/tuple/tuple.tuple/tuple.cnstr/ |
| PR22806_constrain_tuple_like_ctor.pass.cpp | 32 struct IsTuple : std::false_type {};
35 struct IsTuple<Tuple, std::tuple<Args...>> : std::true_type {};
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/tf2xla/ |
| shape_util.cc | 29 if (xla::ShapeUtil::IsTuple(shape)) {
|
| /prebuilts/ndk/r16/sources/cxx-stl/llvm-libc++/test/std/utilities/tuple/tuple.tuple/tuple.cnstr/ |
| PR22806_constrain_tuple_like_ctor.pass.cpp | 32 struct IsTuple : std::false_type {};
35 struct IsTuple<Tuple, std::tuple<Args...>> : std::true_type {};
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/gpu/ |
| infeed_thunk.cc | 42 if (ShapeUtil::IsTuple(hlo_instruction()->shape())) {
|
| gpu_transfer_manager.cc | 55 if (!ShapeUtil::IsTuple(shape)) {
|
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| fused_ir_emitter.cc | 108 if (!ShapeUtil::IsTuple(get_tuple_element->shape())) {
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| client_test.cc | 97 EXPECT_TRUE(ShapeUtil::IsTuple(result->shape()));
|
