Searched refs:num_tuple_inputs (Results 1 – 1 of 1) sorted by relevance
2049 const int num_tuple_inputs) { in MakeBenchmarkWhileBody() argument2052 std::vector<Shape> input_shape(num_tuple_inputs, element_shape); in MakeBenchmarkWhileBody()2056 std::vector<HloInstruction*> gte_nodes(num_tuple_inputs); in MakeBenchmarkWhileBody()2057 for (int i = 0; i < num_tuple_inputs; ++i) { in MakeBenchmarkWhileBody()2066 const int num_tuple_inputs = state.range(0); in BM_ManyElementTuple() local2071 std::vector<HloInstruction*> tuple_params(num_tuple_inputs); in BM_ManyElementTuple()2076 for (int j = 0; j < num_tuple_inputs; ++j) { in BM_ManyElementTuple()2085 module.AddEmbeddedComputation(MakeBenchmarkWhileBody(num_tuple_inputs)); in BM_ManyElementTuple()