Searched refs:DecomposeTuple (Results 1 – 10 of 10) sorted by relevance
486 auto rs = results[replica_idx].DecomposeTuple(); in XLA_TEST_F()828 auto rs = results[replica_idx].DecomposeTuple(); in XLA_TEST_F()
320 auto results = result_tuple.DecomposeTuple(); in XLA_TEST_F()
191 std::vector<Literal> elems = m.DecomposeTuple(); in LiteralToPython()
1725 TEST_F(LiteralUtilTest, DecomposeTuple) { in TEST_F() argument1740 std::vector<Literal> elements = nested_tuple.DecomposeTuple(); in TEST_F()1766 std::vector<Literal> elements = nil_literal.DecomposeTuple(); in TEST_F()
805 std::vector<Literal> DecomposeTuple();
405 std::vector<Literal> Literal::DecomposeTuple() { in DecomposeTuple() function in xla::Literal
795 auto base_elements = base_literal.DecomposeTuple(); in TEST()796 auto nested_0_elements = base_elements[0].Clone().DecomposeTuple(); in TEST()1858 auto exec_literal_parts = exec_literal.DecomposeTuple(); in TEST()1957 auto exec_literal_parts = exec_literal.DecomposeTuple(); in TEST()
4244 std::vector<Literal> actual = literal.DecomposeTuple(); in TEST_F()4401 std::vector<Literal> actual_literals = actual_tuple.DecomposeTuple(); in TEST_F()
2339 std::vector<Literal> computed_results = computed_result.DecomposeTuple(); in PerformReductionStep()
2022 computed_result = computed_result[0].DecomposeTuple();