Searched defs:dimensions_to_reduce (Results 1 – 12 of 12) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/ops/ |
| D | xla_ops.cc | 610 std::vector<int64> dimensions_to_reduce; in __anon592c2f360502() local
660 std::vector<int64> dimensions_to_reduce; in __anon592c2f360702() local
740 std::vector<int64> dimensions_to_reduce; in __anon592c2f360902() local
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | xla_ops_test.py | 474 def reduce(values, *, dimensions_to_reduce): argument
515 def reduce(*values, dimensions_to_reduce): argument
938 def reduce_with_shapes(shape1, shape2, shape3, dimensions_to_reduce=(1,)): argument
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | quantize_and_dequantize_op.cc | 79 std::vector<int64> dimensions_to_reduce; in Compile() local
|
| D | reduction_ops.cc | 133 const std::vector<int64>& dimensions_to_reduce) override { in BuildFinalizer()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ir/ |
| D | mlir_hlo_builder.cc | 171 absl::Span<const int64> dimensions_to_reduce) { in ReduceInternal()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 2454 absl::Span<const int64> dimensions_to_reduce) { in Reduce()
2463 absl::Span<const int64> dimensions_to_reduce) { in Reduce()
2491 absl::Span<const int64> dimensions_to_reduce) { in ReduceInternal()
4436 absl::Span<const int64> dimensions_to_reduce) { in Reduce()
4446 absl::Span<const int64> dimensions_to_reduce) { in Reduce()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_instruction.cc | 1465 absl::Span<const int64> dimensions_to_reduce, in CreateReduce()
1475 absl::Span<const int64> dimensions_to_reduce, in CreateReduce()
|
| D | shape_inference.cc | 2229 absl::Span<const int64> dimensions_to_reduce, in InferReduceShape()
|
| D | hlo_evaluator.cc | 2471 absl::Span<const int64> dimensions_to_reduce(reduce->dimensions()); in HandleReduce() local
|
| D | hlo_parser.cc | 1956 optional<std::vector<int64>> dimensions_to_reduce; in ParseInstructionRhs() local
|
| D | shape_inference_test.cc | 65 absl::Span<const int64> dimensions_to_reduce) { in ExpectInferredReduceShape()
|
| D | hlo_instructions.cc | 1003 absl::Span<const int64> dimensions_to_reduce, in HloReduceInstruction()
|