Home
last modified time | relevance | path

Searched refs:operand_shapes (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_verifier.cc121 std::vector<const Shape*> operand_shapes; in HandleConcatenate() local
123 operand_shapes.push_back(&operand->shape()); in HandleConcatenate()
127 operand_shapes, concatenate->concatenate_dimension())); in HandleConcatenate()
188 std::vector<const Shape*> operand_shapes; in HandleAllReduce() local
190 operand_shapes.push_back(&operand->shape()); in HandleAllReduce()
192 return CheckShape(crs, ShapeInference::InferAllReduceShape(operand_shapes)); in HandleAllReduce()
196 std::vector<const Shape*> operand_shapes; in HandleAllToAll() local
198 operand_shapes.push_back(&operand->shape()); in HandleAllToAll()
201 ShapeInference::InferAllToAllTupleShape(operand_shapes)); in HandleAllToAll()
448 std::vector<const Shape*> operand_shapes; in HandleReduce() local
[all …]
Dshape_inference.cc1023 std::vector<const Shape*> operand_shapes; in InferVariadicOpShape() local
1024 operand_shapes.reserve(operands.size()); in InferVariadicOpShape()
1026 operand_shapes.push_back(&operand->shape()); in InferVariadicOpShape()
1028 return InferVariadicOpShape(opcode, operand_shapes); in InferVariadicOpShape()
1032 HloOpcode opcode, absl::Span<const Shape* const> operand_shapes) { in InferVariadicOpShape() argument
1033 for (const Shape* shape : operand_shapes) { in InferVariadicOpShape()
1039 result.mutable_tuple_shapes()->reserve(operand_shapes.size()); in InferVariadicOpShape()
1040 for (const Shape* shape : operand_shapes) { in InferVariadicOpShape()
1046 if (operand_shapes.size() == 1) { in InferVariadicOpShape()
1047 return *operand_shapes[0]; in InferVariadicOpShape()
[all …]
Dhlo_creation_utils.cc207 std::vector<const Shape*> operand_shapes; in MakeConcatHlo() local
208 absl::c_transform(operands, std::back_inserter(operand_shapes), in MakeConcatHlo()
212 operand_shapes, dimension)); in MakeConcatHlo()
233 std::vector<const Shape*> operand_shapes; in MakeMapHlo() local
237 operand_shapes.push_back(&operand->shape()); in MakeMapHlo()
245 operand_shapes, map_computation->ComputeProgramShape(), map_dims)); in MakeMapHlo()
Dshape_inference.h76 HloOpcode opcode, absl::Span<const Shape* const> operand_shapes);
129 absl::Span<const Shape* const> operand_shapes);
140 absl::Span<const Shape* const> operand_shapes);
Dhlo_evaluator_typed_visitor.h1691 absl::InlinedVector<const Shape*, 1> operand_shapes; in HandleReduce() local
1693 operand_shapes.push_back(&operand->shape()); in HandleReduce()
1697 operand_shapes, in HandleReduce()
Dhlo_instruction.cc448 std::vector<Shape> operand_shapes; in CreateFromProto() local
451 operand_shapes.emplace_back(shape_proto); in CreateFromProto()
455 operand_shapes, proto.custom_call_opaque()); in CreateFromProto()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_autotuning.proto12 repeated xla.ShapeProto operand_shapes = 2; field
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.cc113 std::vector<Shape> operand_shapes; in GetOperandShapes() local
116 operand_shapes.push_back(shape); in GetOperandShapes()
118 return operand_shapes; in GetOperandShapes()
616 TF_ASSIGN_OR_RETURN(const auto& operand_shapes, GetOperandShapes(operands)); in Call()
617 absl::c_transform(operand_shapes, std::back_inserter(operand_shape_ptrs), in Call()
858 TF_ASSIGN_OR_RETURN(const auto& operand_shapes, GetOperandShapes(operands)); in ConcatInDim()
859 absl::c_transform(operand_shapes, std::back_inserter(operand_shape_ptrs), in ConcatInDim()
980 TF_ASSIGN_OR_RETURN(const auto& operand_shapes, GetOperandShapes(elements)); in Tuple()
981 absl::c_transform(operand_shapes, std::back_inserter(operand_shape_ptrs), in Tuple()
1690 TF_ASSIGN_OR_RETURN(std::vector<Shape> operand_shapes, in Sort()
[all …]