Searched defs:arg_dim_steps (Results 1 – 1 of 1) sorted by relevance
3665 absl::Span<const int64_t> arg_dim_steps, in GenerateReduceOutputElement()3750 std::vector<int64_t> arg_dim_steps(arg_dimensions.size()); in HandleReduce() local