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