Home
last modified time | relevance | path

Searched refs:all_operands (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_instruction.cc125 const auto all_operands = [&instruction_map, &proto]() { in CreateFromProto() local
199 instruction = CreateAsyncStart(shape, all_operands(), computations(0), in CreateFromProto()
299 CreateConcatenate(shape, all_operands(), proto.dimensions(0)); in CreateFromProto()
315 auto cond_operands = all_operands(); in CreateFromProto()
330 const auto reduce_operands = all_operands(); in CreateFromProto()
352 auto sort_operands = all_operands(); in CreateFromProto()
353 instruction = CreateSort(shape, proto.dimensions(0), all_operands(), in CreateFromProto()
373 instruction = CreateMap(shape, all_operands(), computations(0)); in CreateFromProto()
423 CreateFusion(shape, fusion_kind, all_operands(), fused_computation); in CreateFromProto()
427 instruction = CreateRng(shape, proto.distribution(), all_operands()); in CreateFromProto()
[all …]
/external/tensorflow/tensorflow/compiler/mlir/xla/ir/
Dmlir_hlo_builder.cc225 const Shape& shape, absl::Span<const XlaOp> all_operands, in ReduceInternal() argument
230 int64_t num_args = all_operands.size() / 2; in ReduceInternal()
232 loc_, GetValues(all_operands.first(num_args)), in ReduceInternal()
233 GetValues(all_operands.subspan(num_args)), in ReduceInternal()
Dmlir_hlo_builder.h154 const Shape& shape, absl::Span<const XlaOp> all_operands,
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.cc2595 std::vector<XlaOp> all_operands; in Reduce() local
2596 all_operands.insert(all_operands.end(), operands.begin(), operands.end()); in Reduce()
2597 all_operands.insert(all_operands.end(), init_values.begin(), in Reduce()
2602 GetOperandShapes(all_operands)); in Reduce()
2610 return ReduceInternal(shape, all_operands, computation, in Reduce()
2616 const Shape& shape, absl::Span<const XlaOp> all_operands, in ReduceInternal() argument
2628 return AddInstruction(std::move(instr), HloOpcode::kReduce, all_operands); in ReduceInternal()
Dxla_builder.h707 const Shape& shape, absl::Span<const XlaOp> all_operands,