Searched defs:dimensions_to_reduce (Results 1 – 10 of 10) sorted by relevance
436 std::vector<int64> dimensions_to_reduce; in __anon00ae58f50402() local
79 std::vector<int64> dimensions_to_reduce; in Compile() local
119 const std::vector<int64>& dimensions_to_reduce) override { in BuildFinalizer()
1907 absl::Span<const int64> dimensions_to_reduce) { in Reduce()1916 absl::Span<const int64> dimensions_to_reduce) { in Reduce()3259 absl::Span<const int64> dimensions_to_reduce) { in Reduce()3269 absl::Span<const int64> dimensions_to_reduce) { in Reduce()
1129 absl::Span<const int64> dimensions_to_reduce, in CreateReduce()1139 absl::Span<const int64> dimensions_to_reduce, in CreateReduce()
2070 absl::Span<const int64> dimensions_to_reduce, in InferReduceShape()
1261 optional<std::vector<int64>> dimensions_to_reduce; in ParseInstructionRhs() local
2387 absl::Span<const int64> dimensions_to_reduce(reduce->dimensions()); in HandleReduce() local
64 absl::Span<const int64> dimensions_to_reduce) { in ExpectInferredReduceShape()
783 absl::Span<const int64> dimensions_to_reduce, in HloReduceInstruction()