Searched defs:dimensions_to_reduce (Results 1 – 9 of 9) sorted by relevance
321 std::vector<int64> dimensions_to_reduce; in __anonc06300730302() local
119 const std::vector<int64>& dimensions_to_reduce) override { in BuildFinalizer()
1940 absl::Span<const int64> dimensions_to_reduce) { in Reduce()1949 absl::Span<const int64> dimensions_to_reduce) { in Reduce()3190 absl::Span<const int64> dimensions_to_reduce) { in Reduce()3200 absl::Span<const int64> dimensions_to_reduce) { in Reduce()
659 absl::Span<const int64> dimensions_to_reduce) { in Reduce()
1034 absl::Span<const int64> dimensions_to_reduce, in CreateReduce()1044 absl::Span<const int64> dimensions_to_reduce, in CreateReduce()
2044 absl::Span<const int64> dimensions_to_reduce, in InferReduceShape()
1199 optional<std::vector<int64>> dimensions_to_reduce; in ParseInstructionRhs() local
63 absl::Span<const int64> dimensions_to_reduce) { in ExpectInferredReduceShape()
729 absl::Span<const int64> dimensions_to_reduce, in HloReduceInstruction()