| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | executable_build_options.cc | 58 const Shape* ExecutableBuildOptions::result_layout() const { in result_layout() function in xla::ExecutableBuildOptions
99 string result_layout = "nullopt"; in ToString() local
101 result_layout = ShapeUtil::HumanStringWithLayout(result_layout_); in ToString()
106 device_ordinal_, result_layout, num_replicas_); in ToString()
116 if (build_options.result_layout() != nullptr) { in CreateExecutionOptions()
118 build_options.result_layout()->ToProto(); in CreateExecutionOptions()
|
| D | compile_only_client.cc | 48 service_instance.result_layout = instance.result_layout; in CompileAheadOfTime()
|
| D | compile_only_client.h | 46 const Shape* result_layout; member
|
| D | executable_build_options.h | 55 const Shape* result_layout() const;
|
| /external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| D | utils.cc | 197 Shape result_layout; in DetermineArgumentLayoutsFromCompileOptions() local
198 if (build_options->result_layout()) { in DetermineArgumentLayoutsFromCompileOptions()
199 result_layout = *build_options->result_layout(); in DetermineArgumentLayoutsFromCompileOptions()
201 result_layout = program_shape.result(); in DetermineArgumentLayoutsFromCompileOptions()
202 LayoutUtil::ClearLayout(&result_layout); in DetermineArgumentLayoutsFromCompileOptions()
204 TF_RETURN_IF_ERROR(assign_layouts(sharded_shapes.second, &result_layout)); in DetermineArgumentLayoutsFromCompileOptions()
205 build_options->set_result_layout(result_layout); in DetermineArgumentLayoutsFromCompileOptions()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | computation_layout.cc | 84 return result_layout() == other.result_layout() && in operator ==()
89 return result_layout() != other.result_layout() || in operator !=()
|
| D | computation_layout.h | 36 explicit ComputationLayout(ShapeLayout result_layout) in ComputationLayout() argument
37 : result_layout_(std::move(result_layout)) {} in ComputationLayout()
67 const ShapeLayout& result_layout() const { return result_layout_; } in result_layout() function
|
| D | layout_assignment.cc | 557 called_computation_layout.result_layout().shape(), instruction)); in AddMandatoryConstraints()
586 if (body_layout.result_layout() != body_layout.parameter_layout(0)) { in AddMandatoryConstraints()
589 << " shape=" << body_layout.result_layout().ToString(); in AddMandatoryConstraints()
590 *body_layout.mutable_parameter_layout(0) = body_layout.result_layout(); in AddMandatoryConstraints()
631 if (!branch_computation_layout.result_layout().MatchesLayoutInShape( in AddMandatoryConstraints()
632 best_branch_computation_layout.result_layout().shape(), in AddMandatoryConstraints()
655 FindOrDie(conditional_mismatch_, computation).result_layout().shape())); in AddMandatoryConstraints()
657 const ShapeLayout& result_layout = computation_layout->result_layout(); in AddMandatoryConstraints() local
658 if (result_layout.LayoutIsSet()) { in AddMandatoryConstraints()
659 TF_RETURN_IF_ERROR(constraints->SetResultLayout(result_layout.shape())); in AddMandatoryConstraints()
[all …]
|
| D | local_service.cc | 141 if (build_options.result_layout() != nullptr) { in CompileExecutables()
142 TF_RETURN_IF_ERROR(ValidateResultShape(*build_options.result_layout(), in CompileExecutables()
|
| D | compile_only_service.h | 45 const Shape* result_layout = nullptr; member
|
| D | compile_only_service.cc | 101 instance.result_layout->ToProto(); in CompileAheadOfTime()
|
| D | hlo_verifier.cc | 1278 const ShapeLayout& result_layout = layout.result_layout(); in VerifyEntryComputationLayout() local
1281 ShapeUtil::ValidateShapeWithOptionalLayout(result_layout.shape())); in VerifyEntryComputationLayout()
1284 result_layout.shape())) { in VerifyEntryComputationLayout()
1289 ShapeUtil::HumanString(result_layout.shape())); in VerifyEntryComputationLayout()
1904 const Layout& result_layout = result_shape.layout(); in Postprocess() local
1910 TF_RET_CHECK(LayoutUtil::Equal(result_layout, operand_layout)) in Postprocess()
|
| D | layout_assignment_test.cc | 244 negate->shape(), computation_layout.result_layout().shape())); in TEST_F()
1299 if (!m->entry_computation_layout().result_layout().LayoutIsSet()) { in AssignLayoutsToComputation()
|
| D | hlo_module.cc | 355 module_config.entry_computation_layout().result_layout().shape(); in CreateFromProto()
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/tests/ |
| D | lhlo_gpu_ops.mlir | 56 result_layout = [3,2,1,0],
69 result_layout = [3,2,1,0],
99 result_layout = [3,2,1,0],
131 result_layout = [3,2,1,0],
162 result_layout = [3,2,1,0],
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | tpu_compile_op_support.cc | 58 absl::optional<const Shape> result_layout, in CreateModuleConfig() argument
88 if (result_layout.has_value()) { in CreateModuleConfig()
90 ValidateResultShape(result_layout.value(), program_shape.result())); in CreateModuleConfig()
93 result_layout.value())); in CreateModuleConfig()
137 absl::optional<const Shape> result_layout, in CreateModuleConfig() argument
140 return CreateModuleConfig(program_shape, argument_shapes, result_layout, in CreateModuleConfig()
|
| D | tpu_compile_op_support.h | 98 absl::optional<const xla::Shape> result_layout,
110 absl::optional<const xla::Shape> result_layout,
|
| D | tpu_util.cc | 61 instance.result_layout = &compilation_result.xla_output_shape; in BuildAotXlaComputationInstance()
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/include/mlir-hlo/Dialect/mhlo/IR/ |
| D | lhlo_gpu_ops_structs.td | 28 // of the first 2 operands of the convolution, and result_layout describes
34 StructFieldAttr<"result_layout", I64ArrayAttr>]> {
|
| /external/tensorflow/tensorflow/stream_executor/tpu/ |
| D | c_api_conversions.cc | 403 xla::ShapeLayout result_layout( in ConfigWithLayout() local
404 FromC(&se_config.entry_computation_layout.result_layout)); in ConfigWithLayout()
405 xla::ComputationLayout layout(result_layout); in ConfigWithLayout()
438 ApiConverter::ToC(layout.result_layout().shape(), in ToC()
439 &hlo_config.entry_computation_layout.result_layout); in ToC()
482 ApiConverter::Free(&c_config->entry_computation_layout.result_layout); in Free()
|
| D | c_api_decl.h | 273 XLA_Shape result_layout; member
|
| /external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/client/ |
| D | tpu_client.cc | 845 ::xla::Shape result_layout; in Compile()
846 if (options.result_layout()) { in Compile()
847 result_layout = *options.result_layout(); in Compile()
858 result_layout = ::xla::Shape(program_shape_proto.result()); in Compile()
860 VLOG(1) << "Got result shape: " << result_layout.DebugString(); in Compile()
864 std::move(client), std::move(result_layout), tuple_arguments); in Compile()
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | xla_compiler.cc | 532 return options.result_layout() in BuildXlaCompilerSubmodule()
533 ? absl::optional<Shape>(*options.result_layout()) in BuildXlaCompilerSubmodule()
|
| /external/tensorflow/tensorflow/compiler/aot/ |
| D | compile.cc | 89 instance.result_layout = &result_shape; in CompileXla()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/tests/hlo_to_lhlo_with_xla/ |
| D | hlo_text_to_lhlo_no_opt.hlotxt | 281 // CHECK-SAME: result_layout = [3, 2, 1, 0]
318 // CHECK-SAME: result_layout = [1, 3, 2, 0]
357 // CHECK-SAME: result_layout = [1, 3, 2, 0]
|