Searched defs:operand_shapes_with_layout (Results 1 – 7 of 7) sorted by relevance
137 const py::bytes& opaque, bool has_side_effect) -> XlaOp { in BuildOpsSubmodule()
211 repeated xla.ShapeProto operand_shapes_with_layout = 57; field
1487 const std::vector<Shape>& operand_shapes_with_layout() const { in operand_shapes_with_layout() function
1599 absl::Span<const Shape> operand_shapes_with_layout, string opaque) { in CreateCustomCall()
2395 absl::Span<const Shape> operand_shapes_with_layout) in HloCustomCallInstruction()
134 absl::optional<absl::Span<const Shape>> operand_shapes_with_layout, in CustomCallInternal()
1882 absl::optional<absl::Span<const Shape>> operand_shapes_with_layout, in CustomCall()1926 absl::optional<absl::Span<const Shape>> operand_shapes_with_layout, in CustomCallInternal()1961 absl::optional<absl::Span<const Shape>> operand_shapes_with_layout, in CustomCall()4491 absl::Span<const Shape> operand_shapes_with_layout, const string& opaque, in CustomCallWithLayout()