Searched refs:arg_dim_steps (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_evaluator.cc | 3665 absl::Span<const int64_t> arg_dim_steps, in GenerateReduceOutputElement() argument
3692 arg_shape, base, arg_dim_counts, arg_dim_steps, reduction_step)); in GenerateReduceOutputElement()
3700 arg_shape, base, arg_dim_counts, arg_dim_steps, in GenerateReduceOutputElement()
3750 std::vector<int64_t> arg_dim_steps(arg_dimensions.size()); in HandleReduce() local
3759 arg_dim_steps[dim] = 1; in HandleReduce()
3767 if (arg_dim_steps[i] == 0) { in HandleReduce()
3784 arg_dim_steps, arg_dim_counts, result_to_arg_index); in HandleReduce()
|