Home
last modified time | relevance | path

Searched refs:tuple_element_count (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_module_dce.cc65 const int64_t tuple_element_count = in RunWhileDCE() local
68 for (int64_t i = 0; i < tuple_element_count; ++i) { in RunWhileDCE()
H A Dconditional_simplifier.cc559 const int64_t tuple_element_count = in TryRemoveConditional() local
561 selects.reserve(tuple_element_count); in TryRemoveConditional()
562 for (int64_t i = 0; i < tuple_element_count; ++i) { in TryRemoveConditional()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xrt/
H A Dxrt_util.cc353 int64_t tuple_element_count = in CreateExecuteOutput() local
357 0, TensorShape({tuple_element_count}), &output_tensor)); in CreateExecuteOutput()
359 for (int64_t i = 0; i < tuple_element_count; ++i) { in CreateExecuteOutput()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/
H A Dliteral.cc524 const auto tuple_element_count = ShapeUtil::TupleElementCount(shape()); in DecomposeTuple() local
525 elements.reserve(tuple_element_count); in DecomposeTuple()
526 for (int i = 0; i < tuple_element_count; ++i) { in DecomposeTuple()
1307 const auto tuple_element_count = ShapeUtil::TupleElementCount(subshape); in TupleToStringHelper() local
1308 tuple_pieces.reserve(tuple_element_count); in TupleToStringHelper()
1736 const auto tuple_element_count = ShapeUtil::TupleElementCount(shape()); in ConvertToShape() local
1737 elements.reserve(tuple_element_count); in ConvertToShape()
1738 for (int i = 0; i < tuple_element_count; ++i) { in ConvertToShape()