Searched defs:operand_shapes_with_layout (Results 1 – 9 of 9) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/python/ |
| H A D | ops.cc | 195 absl::Span<const Shape> operand_shapes_with_layout, in BuildOpsSubmodule()
200 CustomCallApiVersion api_version) -> XlaOp { in BuildOpsSubmodule()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| H A D | light_outside_compilation.cc | 103 std::vector<xla::Shape> operand_shapes_with_layout; in CompileToCustomCallCallingTfKernel() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | hlo.proto | 266 repeated xla.ShapeProto operand_shapes_with_layout = 57; field
|
| H A D | hlo_instructions.h | 1890 const std::vector<Shape>& operand_shapes_with_layout() const { in operand_shapes_with_layout() function
|
| H A D | hlo_instruction.cc | 724 const auto& operand_shapes_with_layout = in CreateFromProto() local
1908 absl::Span<const Shape> operand_shapes_with_layout, std::string opaque, in CreateCustomCall()
|
| H A D | hlo_instructions.cc | 2696 absl::Span<const Shape> operand_shapes_with_layout, in HloCustomCallInstruction()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/ |
| H A D | xla_builder.cc | 1968 std::optional<absl::Span<const Shape>> operand_shapes_with_layout, in CustomCall()
2016 std::optional<absl::Span<const Shape>> operand_shapes_with_layout, in CustomCallInternal()
2077 std::optional<absl::Span<const Shape>> operand_shapes_with_layout, in CustomCall()
4452 absl::Span<const Shape> operand_shapes_with_layout, in CustomCallWithLayout()
4467 absl::Span<const Shape> operand_shapes_with_layout, in CustomCallWithConvDnums()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/ir/ |
| H A D | mlir_hlo_builder.cc | 146 std::optional<absl::Span<const Shape>> operand_shapes_with_layout, in CustomCallInternal()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/ |
| H A D | mlir_hlo_to_hlo.cc | 1132 auto operand_shapes_with_layout = ConvertTypesToShapesWithLayout( in ExportXlaOp() local
|