Searched defs:operand_shapes_with_layout (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | ops.cc | 176 absl::Span<const Shape> operand_shapes_with_layout, in BuildOpsSubmodule()
181 CustomCallApiVersion api_version) -> XlaOp { in BuildOpsSubmodule()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo.proto | 246 repeated xla.ShapeProto operand_shapes_with_layout = 57; field
|
| D | hlo_instructions.h | 1572 const std::vector<Shape>& operand_shapes_with_layout() const { in operand_shapes_with_layout() function
|
| D | hlo_instruction.cc | 1785 absl::Span<const Shape> operand_shapes_with_layout, string opaque, in CreateCustomCall()
|
| D | hlo_instructions.cc | 2542 absl::Span<const Shape> operand_shapes_with_layout, in HloCustomCallInstruction()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ir/ |
| D | mlir_hlo_builder.cc | 134 absl::optional<absl::Span<const Shape>> operand_shapes_with_layout, in CustomCallInternal()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 1894 absl::optional<absl::Span<const Shape>> operand_shapes_with_layout, in CustomCall()
1941 absl::optional<absl::Span<const Shape>> operand_shapes_with_layout, in CustomCallInternal()
1986 absl::optional<absl::Span<const Shape>> operand_shapes_with_layout, in CustomCall()
4307 absl::Span<const Shape> operand_shapes_with_layout, const string& opaque, in CustomCallWithLayout()
4322 absl::Span<const Shape> operand_shapes_with_layout, const string& opaque, in CustomCallWithConvDnums()
|