Searched refs:num_tuple_inputs (Results 1 – 1 of 1) sorted by relevance
1993 const int num_tuple_inputs) { in MakeBenchmarkWhileBody() argument1996 std::vector<Shape> input_shape(num_tuple_inputs, element_shape); in MakeBenchmarkWhileBody()2000 std::vector<HloInstruction*> gte_nodes(num_tuple_inputs); in MakeBenchmarkWhileBody()2001 for (int i = 0; i < num_tuple_inputs; ++i) { in MakeBenchmarkWhileBody()2009 void BM_ManyElementTuple(int num_iters, const int num_tuple_inputs) { in BM_ManyElementTuple() argument2015 std::vector<HloInstruction*> tuple_params(num_tuple_inputs); in BM_ManyElementTuple()2019 for (int j = 0; j < num_tuple_inputs; ++j) { in BM_ManyElementTuple()2028 module.AddEmbeddedComputation(MakeBenchmarkWhileBody(num_tuple_inputs)); in BM_ManyElementTuple()