| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_sharding.cc | 127 CHECK(tuple_shape.IsTuple()) << ShapeUtil::HumanString(tuple_shape); in Tuple()
129 CHECK(!sharding.IsTuple()) << sharding.ToString(); in Tuple()
140 CHECK(tuple_shape.IsTuple()) << ShapeUtil::HumanString(tuple_shape); in SingleTuple()
141 CHECK(!sharding.IsTuple()) << sharding.ToString(); in SingleTuple()
150 return shape.IsTuple() ? SingleTuple(shape, sharding) : sharding; in Single()
154 if (IsTuple()) { in ToString()
198 if (IsTuple()) { in UsesDevice()
210 if (IsTuple()) { in UsedDevices()
233 CHECK(!IsTuple()); in TileIndexForDevice()
250 CHECK(!IsTuple()); in DeviceForTileIndex()
[all …]
|
| D | root_instruction_sinker.cc | 26 CHECK(root->shape().IsTuple()); in SinkTupleRoot()
41 CHECK(!root->shape().IsTuple()); in SinkNontupleRoot()
63 if (computation->root_instruction()->shape().IsTuple()) { in Run()
|
| D | optimize_input_output_buffer_alias.cc | 35 return shape.IsTuple() && !ShapeUtil::IsNestedTuple(shape); in IsNonNestedTuple()
58 if (subshape.IsTuple()) { in Build()
68 if (subshape.IsTuple()) { in Build()
|
| D | conditional_canonicalizer.cc | 28 TF_RET_CHECK(!root->shape().IsTuple()); in CanonicalizeNonTupleConditional()
53 !inst->shape().IsTuple()) { in Run()
|
| D | tuple_util.cc | 24 CHECK(input_tuple->shape().IsTuple()); in ExtractPrefix()
44 CHECK(input_tuple->shape().IsTuple()); in AppendSuffix()
|
| D | hlo_sharding.h | 123 bool IsTuple() const { return tuple_; } in IsTuple() function
127 if (!IsTuple()) { in IsReplicated()
136 if (!IsTuple()) { in IsTileMaximal()
146 if (!IsTuple()) { in IsManual()
|
| D | hlo_sharding_metadata.cc | 188 TF_RET_CHECK(!lhs->IsTuple() && !rhs.IsTuple()); in AssignLeafSharding()
237 TF_RET_CHECK(instruction->shape().IsTuple() && in ApplyShardingFromUsers()
269 if (instruction->shape().IsTuple()) { in ApplyShardingFromUsers()
301 if (instruction->shape().IsTuple()) { in ApplyShardingFromUsers()
364 TF_RET_CHECK(instruction->shape().IsTuple()) in ApplyDomainSharding()
|
| D | hlo_element_type_converter.cc | 71 CHECK(!subshape.IsTuple()); in GetConvertedTupleShape()
95 CHECK(!ele_shape.IsTuple()); in ConvertTupleElements()
194 } else if (hlo->shape().IsTuple()) { in Run()
|
| D | hlo_query.cc | 156 if (send->is_host_transfer() && send->operand(0)->shape().IsTuple()) { in HasX64TransformedHostTransfer()
162 recv->shape().tuple_shapes(0).IsTuple()) { in HasX64TransformedHostTransfer()
|
| D | hlo_value.cc | 50 instruction->shape().IsTuple() ? (" " + index.ToString()) : ""; in ToString()
60 string index_str = instruction->operand(operand_number)->shape().IsTuple() in ToString()
93 instruction()->shape().IsTuple() ? index().ToString() : "", in ToShortString()
|
| D | sort_simplifier.cc | 34 if (!sort->shape().IsTuple()) { in RemoveUnusedOperandFromSort()
112 if (new_sort->shape().IsTuple()) { in RemoveUnusedOperandFromSort()
|
| D | transfer_manager.cc | 215 if (buffer_shape.IsTuple()) { in ReadDynamicShapes()
300 if (device_subshape.IsTuple() && in WriteTupleIndexTablesAsync()
324 TF_RET_CHECK(device_buffer.on_device_shape().IsTuple()); in WriteRootTupleIndexTable()
343 TF_RET_CHECK(buffer_tree.shape().IsTuple()); in WriteRootTupleIndexTable()
|
| D | conditional_simplifier.cc | 210 if (!param->shape().IsTuple()) { in TryRemoveUnusedConditionalOperands()
342 !conditional_op->shape().IsTuple()) { in RemoveUnusedTupleElements()
393 if (!root->shape().IsTuple() || in RemoveUnusedTupleElements()
494 !conditional->shape().IsTuple()) { in MergeDuplicateTupleElements()
|
| D | while_loop_simplifier.cc | 229 if (!while_init->shape().IsTuple()) { in TryRemoveDeadWhileParams()
403 if (!while_init->shape().IsTuple()) { in TryRemoveRepeatedWhileTupleIndices()
410 if (!while_init->shape().IsTuple() || in TryRemoveRepeatedWhileTupleIndices()
797 CHECK(desired_shape.IsTuple()) << ShapeUtil::HumanString(desired_shape); in UnflattenTupleInstr()
806 if (!subshape.IsTuple()) { in UnflattenTupleInstr()
816 if (!s.IsTuple()) { in UnflattenTupleInstr()
838 if (!shape.IsTuple()) { in GetFlatTupleElems()
878 if (!s.IsTuple()) { in TryFlattenNestedTuples()
|
| D | buffer_value.cc | 33 is_tuple_ = shape.IsTuple(); in BufferValue()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | layout_util.cc | 145 if (shape->IsTuple()) { in SetToDefaultLayout()
177 if (shape.IsTuple()) { in ValidateLayoutInShape()
209 if (shape.IsTuple()) { in ValidateLayoutForShape()
296 if (shape.IsTuple()) { in HasLayout()
367 if (src.IsTuple() != dst->IsTuple()) { in CopyLayoutInternal()
371 if (src.IsTuple()) { in CopyLayoutInternal()
405 if (lhs.IsTuple()) { in LayoutsInShapesEqual()
406 if (!rhs.IsTuple() || ShapeUtil::TupleElementCount(lhs) != in LayoutsInShapesEqual()
|
| D | shape_layout.cc | 60 CHECK(!shape_.IsTuple()); in layout()
69 CHECK(!shape_.IsTuple()); in ResetLayout()
77 CHECK(shape_.IsTuple()); in ResetLayout()
|
| D | shape_util.cc | 407 CHECK(!shape->IsTuple()); in UpdateDynamicDimension()
486 return shape.IsTuple() && in IsNestedTuple()
488 [](const Shape& s) { return s.IsTuple(); }); in IsNestedTuple()
492 return shape.IsTuple() && TupleElementCount(shape) == 0; in IsEmptyTuple()
496 CHECK(shape.IsTuple()) << HumanString(shape); in TupleElementCount()
502 CHECK(shape.IsTuple()); in GetTupleElementShape()
518 CHECK(tuple.IsTuple()); in SliceTuple()
547 CHECK(shape.IsArray() || shape.IsTuple()); in ElementsInRecursive()
581 if (shape.IsTuple()) { in HumanString()
605 if (shape.IsTuple()) { in HumanStringWithLayout()
[all …]
|
| D | shape.cc | 87 if (IsTuple()) { in is_static()
120 if (lhs.IsTuple()) { in operator ()()
121 return rhs.IsTuple() && in operator ()()
|
| /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/llvm-project/libcxx/test/std/utilities/tuple/tuple.tuple/tuple.cnstr/ |
| D | PR22806_constrain_tuple_like_ctor.pass.cpp | 33 struct IsTuple : std::false_type {}; struct
36 struct IsTuple<Tuple, std::tuple<Args...>> : std::true_type {}; struct
45 template <class Tuple, class = typename std::enable_if<IsTuple<Tuple>::value>::type>
58 template <class Tuple, class = typename std::enable_if<IsTuple<Tuple>::value>::type>
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | shape_util.cc | 30 if (shape.IsTuple()) { in PopulateInfeedLayoutVector()
90 if (shape.IsTuple()) { in XLAShapeToTensorShape()
177 if (input_shape.IsTuple()) { in GetShapeWithLayout()
185 if (shape.IsTuple()) { in GetShapeWithLayout()
|
| /external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/client/ |
| D | tpu_client.cc | 184 if (!tuple_shape.IsTuple()) { in FromLiterals()
309 if (!on_host_shape_.IsTuple()) { in CopyToHostAsync()
368 if (!on_host_shape_.IsTuple()) { in DestructureTuple()
393 if (on_host_shape_.IsTuple()) { in CopyToDevice()
446 if (!shape.IsTuple()) { in AllocateBuffer()
478 TF_RET_CHECK(!non_tuple_shape.IsTuple()); in CreateBuffer()
657 if (result.buffer->on_host_shape().IsTuple()) { in Execute()
744 if (results[i].buffer->on_host_shape().IsTuple()) { in ExecuteOnLocalDevices()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_nested.cc | 138 CHECK(return_shape.IsTuple()); in CodegenNestedComputation()
171 if (hlo.shape().IsTuple()) { in EmitTargetElementLoop()
|
| /external/tensorflow/tensorflow/python/util/ |
| D | util.h | 107 bool IsTuple(PyObject* o);
|