Searched defs:operand_shapes_with_layout (Results 1 – 7 of 7) sorted by relevance
202 repeated xla.ShapeProto operand_shapes_with_layout = 57; field
1219 const std::vector<Shape>& operand_shapes_with_layout() const { in operand_shapes_with_layout() function
1290 absl::Span<const Shape> operand_shapes_with_layout, in CreateCustomCall()
2061 absl::Span<const Shape> operand_shapes_with_layout) in HloCustomCallInstruction()
621 const std::vector<Shape>& operand_shapes_with_layout, in CustomCall()
1422 operand_shapes_with_layout, argument
1496 absl::optional<absl::Span<const Shape>> operand_shapes_with_layout) { in CustomCall()3090 absl::Span<const Shape> operand_shapes_with_layout, in CustomCallWithLayout()