Searched defs:dimensions_to_reduce (Results 1 – 13 of 13) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/ops/ |
| H A D | xla_ops.cc | 613 std::vector<int64_t> dimensions_to_reduce; in __anon6b9e6b800502() local
663 std::vector<int64_t> dimensions_to_reduce; in __anon6b9e6b800702() local
743 std::vector<int64_t> dimensions_to_reduce; in __anon6b9e6b800902() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tests/ |
| H A D | xla_ops_test.py | 489 def reduce(values, *, dimensions_to_reduce): argument
539 def reduce(*values, dimensions_to_reduce): argument
993 def reduce_with_shapes(shape1, shape2, shape3, dimensions_to_reduce=(1,)): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| H A D | ir_emission_utils.cc | 259 llvm::SmallVector<int64_t> dimensions_to_reduce; in IsReductionFromOrToContiguousDimensions() local
303 llvm::SmallVector<int64_t> dimensions_to_reduce; in GetReductionKindAndContiguousComponents() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| H A D | quantize_and_dequantize_op.cc | 79 std::vector<int64_t> dimensions_to_reduce; in Compile() local
|
| H A D | reduction_ops.cc | 133 const std::vector<int64_t>& dimensions_to_reduce) override { in BuildFinalizer()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/ |
| H A D | xla_builder.cc | 2581 absl::Span<const int64_t> dimensions_to_reduce) { in Reduce()
2590 absl::Span<const int64_t> dimensions_to_reduce) { in Reduce()
2618 absl::Span<const int64_t> dimensions_to_reduce) { in ReduceInternal()
4585 absl::Span<const int64_t> dimensions_to_reduce) { in Reduce()
4595 absl::Span<const int64_t> dimensions_to_reduce) { in Reduce()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | hlo_instruction.cc | 1563 absl::Span<const int64_t> dimensions_to_reduce, in CreateReduce()
1573 absl::Span<const int64_t> dimensions_to_reduce, in CreateReduce()
1586 absl::Span<const int64_t> dimensions_to_reduce, in CreateReduce()
|
| H A D | shape_inference.cc | 2218 absl::Span<const int64_t> dimensions_to_reduce, in InferReduceShape()
|
| H A D | hlo_parser.cc | 2186 optional<std::vector<int64_t>> dimensions_to_reduce; in CreateInstruction() local
|
| H A D | hlo_evaluator.cc | 3712 absl::Span<const int64_t> dimensions_to_reduce(reduce->dimensions()); in HandleReduce() local
|
| H A D | hlo_instructions.cc | 1149 absl::Span<const int64_t> dimensions_to_reduce, in HloReduceInstruction()
|
| H A D | shape_inference_test.cc | 67 absl::Span<const int64_t> dimensions_to_reduce) { in ExpectInferredReduceShape()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/ir/ |
| H A D | mlir_hlo_builder.cc | 227 absl::Span<const int64_t> dimensions_to_reduce) { in ReduceInternal()
|