Searched defs:dimensions_to_reduce (Results 1 – 11 of 11) sorted by relevance
537 std::vector<int64> dimensions_to_reduce; in __anonbbc1c2d10502() local587 std::vector<int64> dimensions_to_reduce; in __anonbbc1c2d10702() local
79 std::vector<int64> dimensions_to_reduce; in Compile() local
133 const std::vector<int64>& dimensions_to_reduce) override { in BuildFinalizer()
158 absl::Span<const int64> dimensions_to_reduce) { in ReduceInternal()
2426 absl::Span<const int64> dimensions_to_reduce) { in Reduce()2435 absl::Span<const int64> dimensions_to_reduce) { in Reduce()2463 absl::Span<const int64> dimensions_to_reduce) { in ReduceInternal()4599 absl::Span<const int64> dimensions_to_reduce) { in Reduce()4609 absl::Span<const int64> dimensions_to_reduce) { in Reduce()
1288 absl::Span<const int64> dimensions_to_reduce, in CreateReduce()1298 absl::Span<const int64> dimensions_to_reduce, in CreateReduce()
2115 absl::Span<const int64> dimensions_to_reduce, in InferReduceShape()
2407 absl::Span<const int64> dimensions_to_reduce(reduce->dimensions()); in HandleReduce() local
1819 optional<std::vector<int64>> dimensions_to_reduce; in ParseInstructionRhs() local
65 absl::Span<const int64> dimensions_to_reduce) { in ExpectInferredReduceShape()
903 absl::Span<const int64> dimensions_to_reduce, in HloReduceInstruction()