| /external/tensorflow/tensorflow/compiler/tf2xla/ops/ |
| D | xla_ops.cc | 321 std::vector<int64> dimensions_to_reduce; in __anonc06300730302() local
323 c->GetAttr("dimensions_to_reduce", &dimensions_to_reduce)); in __anonc06300730302()
324 std::set<int64> dims_set(dimensions_to_reduce.begin(), in __anonc06300730302()
325 dimensions_to_reduce.end()); in __anonc06300730302()
329 if (rank < dimensions_to_reduce.size() || in __anonc06300730302()
330 dims_set.size() != dimensions_to_reduce.size() || in __anonc06300730302()
331 !absl::c_all_of(dimensions_to_reduce, dim_in_range)) { in __anonc06300730302()
336 0, c->UnknownShapeOfRank(rank - dimensions_to_reduce.size())); in __anonc06300730302()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | reduction_ops.cc | 119 const std::vector<int64>& dimensions_to_reduce) override { in BuildFinalizer() argument
120 if (dimensions_to_reduce.empty()) { in BuildFinalizer()
123 auto divisor = xla::GetDimensionSize(input, dimensions_to_reduce[0]); in BuildFinalizer()
124 for (int i = 1; i < dimensions_to_reduce.size(); i++) { in BuildFinalizer()
125 auto size = xla::GetDimensionSize(input, dimensions_to_reduce[i]); in BuildFinalizer()
|
| D | reduction_ops.h | 58 const std::vector<int64>& dimensions_to_reduce);
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | xla_ops_test.py | 221 x, init_value=0, dimensions_to_reduce=dims, reducer=sum_reducer)
250 x, init_value=1, dimensions_to_reduce=dims, reducer=mul_reducer)
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.h | 442 absl::Span<const int64> dimensions_to_reduce);
447 absl::Span<const int64> dimensions_to_reduce);
864 absl::Span<const int64> dimensions_to_reduce);
868 absl::Span<const int64> dimensions_to_reduce);
1533 absl::Span<const int64> dimensions_to_reduce);
1540 absl::Span<const int64> dimensions_to_reduce);
|
| D | xla_builder.cc | 1940 absl::Span<const int64> dimensions_to_reduce) { in Reduce() argument
1943 dimensions_to_reduce); in Reduce()
1949 absl::Span<const int64> dimensions_to_reduce) { in Reduce() argument
1970 operand_shape_ptrs, dimensions_to_reduce, called_program_shape)); in Reduce()
1973 for (int64 dim : dimensions_to_reduce) { in Reduce()
3190 absl::Span<const int64> dimensions_to_reduce) { in Reduce() argument
3192 dimensions_to_reduce); in Reduce()
3200 absl::Span<const int64> dimensions_to_reduce) { in Reduce() argument
3202 dimensions_to_reduce); in Reduce()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | shape_inference.h | 153 absl::Span<const int64> dimensions_to_reduce,
|
| D | hlo_instruction.h | 610 absl::Span<const int64> dimensions_to_reduce,
626 absl::Span<const int64> dimensions_to_reduce,
|
| D | shape_inference.cc | 2044 absl::Span<const int64> dimensions_to_reduce, in InferReduceShape() argument
2073 for (int64 dimension : dimensions_to_reduce) { in InferReduceShape()
2088 std::set<int64> dimensions_to_reduce_set(dimensions_to_reduce.begin(), in InferReduceShape()
2089 dimensions_to_reduce.end()); in InferReduceShape()
|
| D | hlo_instruction.cc | 1034 absl::Span<const int64> dimensions_to_reduce, in CreateReduce() argument
1037 shape, {operand, init_value}, dimensions_to_reduce, reduce_computation)); in CreateReduce()
1044 absl::Span<const int64> dimensions_to_reduce, in CreateReduce() argument
1051 shape, all_args, dimensions_to_reduce, reduce_computation); in CreateReduce()
|
| D | hlo_instructions.cc | 729 absl::Span<const int64> dimensions_to_reduce, in HloReduceInstruction() argument
732 dimensions_(dimensions_to_reduce.begin(), dimensions_to_reduce.end()) { in HloReduceInstruction()
|
| D | hlo_parser.cc | 1199 optional<std::vector<int64>> dimensions_to_reduce; in ParseInstructionRhs() local
1201 &dimensions_to_reduce}; in ParseInstructionRhs()
1216 *dimensions_to_reduce, *reduce_computation)); in ParseInstructionRhs()
|
| D | shape_inference_test.cc | 63 absl::Span<const int64> dimensions_to_reduce) { in ExpectInferredReduceShape() argument
66 {&arg, &f32_}, dimensions_to_reduce, to_apply); in ExpectInferredReduceShape()
|
| D | hlo_instructions.h | 456 absl::Span<const int64> dimensions_to_reduce,
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | local_computation_builder.cc | 659 absl::Span<const int64> dimensions_to_reduce) { in Reduce() argument
661 local_computation.computation(), dimensions_to_reduce); in Reduce()
|
| D | local_computation_builder.h | 328 absl::Span<const int64> dimensions_to_reduce);
|