Searched refs:all_operands (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_instruction.cc | 102 const auto all_operands = [&instruction_map, &proto]() { in CreateFromProto() local
212 CreateConcatenate(shape, all_operands(), proto.dimensions(0)); in CreateFromProto()
228 auto cond_operands = all_operands(); in CreateFromProto()
243 const auto reduce_operands = all_operands(); in CreateFromProto()
265 auto sort_operands = all_operands(); in CreateFromProto()
266 instruction = CreateSort(shape, proto.dimensions(0), all_operands(), in CreateFromProto()
286 instruction = CreateMap(shape, all_operands(), computations(0)); in CreateFromProto()
336 CreateFusion(shape, fusion_kind, all_operands(), fused_computation); in CreateFromProto()
340 instruction = CreateRng(shape, proto.distribution(), all_operands()); in CreateFromProto()
384 shape, all_operands(), computations(0), in CreateFromProto()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 1956 std::vector<XlaOp> all_operands; in Reduce() local
1957 all_operands.insert(all_operands.end(), operands.begin(), operands.end()); in Reduce()
1958 all_operands.insert(all_operands.end(), init_values.begin(), in Reduce()
1963 GetOperandShapes(all_operands)); in Reduce()
1979 return AddInstruction(std::move(instr), HloOpcode::kReduce, all_operands); in Reduce()
|