| /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
132 operand_shapes_with_layout.push_back(xla_shape); in CompileToCustomCallCallingTfKernel()
133 xla::LayoutUtil::SetToDefaultLayout(&operand_shapes_with_layout.back()); in CompileToCustomCallCallingTfKernel()
219 /*operand_shapes_with_layout=*/operand_shapes_with_layout, in CompileToCustomCallCallingTfKernel()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/python/ |
| H A D | ops.cc | 176 absl::Span<const Shape> operand_shapes_with_layout, in BuildOpsSubmodule()
182 operand_shapes_with_layout, opaque, has_side_effect, in BuildOpsSubmodule()
195 absl::Span<const Shape> operand_shapes_with_layout, in BuildOpsSubmodule() argument
203 operand_shapes_with_layout, opaque, has_side_effect, in BuildOpsSubmodule()
|
| H A D | xla_client_test.py | 381 operand_shapes_with_layout=[
406 operand_shapes_with_layout=[
|
| /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() argument
1982 if (operand_shapes_with_layout.has_value()) { in CustomCall()
1988 if (operands.size() != operand_shapes_with_layout->size()) { in CustomCall()
1992 operand_shapes_with_layout->size(), operands.size()); in CustomCall()
1995 for (const Shape& operand_shape : *operand_shapes_with_layout) { in CustomCall()
2007 operand_shapes_with_layout, has_side_effect, output_operand_aliasing, in CustomCall()
2016 std::optional<absl::Span<const Shape>> operand_shapes_with_layout, in CustomCallInternal() argument
2039 if (operand_shapes_with_layout.has_value()) { in CustomCallInternal()
2041 for (const Shape& operand_shape : *operand_shapes_with_layout) { in CustomCallInternal()
2077 std::optional<absl::Span<const Shape>> operand_shapes_with_layout, in CustomCall() argument
[all …]
|
| H A D | xla_builder.h | 659 std::optional<absl::Span<const Shape>> operand_shapes_with_layout,
674 std::optional<absl::Span<const Shape>> operand_shapes_with_layout,
688 std::optional<absl::Span<const Shape>> operand_shapes_with_layout,
1278 absl::Span<const Shape> operand_shapes_with_layout,
1287 absl::Span<const Shape> operand_shapes_with_layout,
2161 absl::Span<const Shape> operand_shapes_with_layout,
2178 absl::Span<const Shape> operand_shapes_with_layout,
|
| /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() argument
169 if (operand_shapes_with_layout.has_value()) { in CustomCallInternal()
172 operand_shapes_with_layout.value(), &builder_)); in CustomCallInternal()
|
| H A D | mlir_hlo_builder.h | 145 std::optional<absl::Span<const Shape>> operand_shapes_with_layout,
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/python/xla_extension/ |
| H A D | ops.pyi | 187 operand_shapes_with_layout: Sequence[Shape],
197 operand_shapes_with_layout: Sequence[Shape],
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | hlo_instruction.cc | 724 const auto& operand_shapes_with_layout = in CreateFromProto() local
725 proto.operand_shapes_with_layout(); in CreateFromProto()
726 operand_shapes.reserve(operand_shapes_with_layout.size()); in CreateFromProto()
727 for (const ShapeProto& shape_proto : operand_shapes_with_layout) { in CreateFromProto()
1908 absl::Span<const Shape> operand_shapes_with_layout, std::string opaque, in CreateCustomCall() argument
1912 operand_shapes_with_layout, api_version); in CreateCustomCall()
|
| H A D | hlo.proto | 263 // 'operand_shapes_with_layout' must contain a shape with layout for each
266 repeated xla.ShapeProto operand_shapes_with_layout = 57; field
|
| H A D | hlo_instructions.cc | 2696 absl::Span<const Shape> operand_shapes_with_layout, in HloCustomCallInstruction() argument
2704 operand_shapes_with_layout_(operand_shapes_with_layout.begin(), in HloCustomCallInstruction()
2705 operand_shapes_with_layout.end()), in HloCustomCallInstruction()
2911 cloned->operand_shapes_with_layout_ = operand_shapes_with_layout(); in CloneWithNewOperandsImpl()
|
| H A D | hlo_instructions.h | 1806 absl::Span<const Shape> operand_shapes_with_layout,
1890 const std::vector<Shape>& operand_shapes_with_layout() const { in operand_shapes_with_layout() function
|
| H A D | hlo_verifier.cc | 1172 custom_call->operand_shapes_with_layout().size()); in HandleCustomCall()
1175 custom_call->operand_shapes_with_layout()[i]; in HandleCustomCall()
|
| H A D | layout_assignment.cc | 634 custom_call->operand_shapes_with_layout()[i], custom_call, i)); in AddMandatoryConstraints()
851 custom_call->operand_shapes_with_layout()[i])); in CheckCustomCallLayout()
|
| H A D | hlo_instruction.h | 1137 absl::Span<const Shape> operand_shapes_with_layout,
|
| /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
1138 result_shape_with_layout, operand_shapes_with_layout, in ExportXlaOp()
|
| H A D | hlo_function_importer.cc | 632 ExtractLayoutsFromShapes(custom_call->operand_shapes_with_layout(), in ImportInstructionImpl()
|