Searched defs:all_operands (Results 1 – 3 of 3) sorted by relevance
156 const Shape& shape, absl::Span<const XlaOp> all_operands, in ReduceInternal()
2440 std::vector<XlaOp> all_operands; in Reduce() local2461 const Shape& shape, absl::Span<const XlaOp> all_operands, in ReduceInternal()
108 const auto all_operands = [&instruction_map, &proto]() { in CreateFromProto() local