| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | executable_build_options.cc | 57 const Shape* ExecutableBuildOptions::result_layout() const { in result_layout() function in xla::ExecutableBuildOptions
74 string result_layout = "nullopt"; in ToString() local
76 result_layout = ShapeUtil::HumanStringWithLayout(result_layout_); in ToString()
81 device_ordinal_, result_layout, num_replicas_); in ToString()
|
| D | compile_only_client.cc | 47 service_instance.result_layout = instance.result_layout; in CompileAheadOfTime()
|
| D | executable_build_options.h | 47 const Shape* result_layout() const;
|
| D | compile_only_client.h | 46 const Shape* result_layout; member
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | computation_layout.cc | 71 return result_layout() == other.result_layout() && in operator ==()
76 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 | 556 called_computation_layout.result_layout().shape(), instruction)); in AddMandatoryConstraints()
585 if (body_layout.result_layout() != body_layout.parameter_layout(0)) { in AddMandatoryConstraints()
588 << " shape=" << body_layout.result_layout().ToString(); in AddMandatoryConstraints()
589 *body_layout.mutable_parameter_layout(0) = body_layout.result_layout(); in AddMandatoryConstraints()
630 if (!branch_computation_layout.result_layout().MatchesLayoutInShape( in AddMandatoryConstraints()
631 best_branch_computation_layout.result_layout().shape(), in AddMandatoryConstraints()
654 FindOrDie(conditional_mismatch_, computation).result_layout().shape())); in AddMandatoryConstraints()
656 const ShapeLayout& result_layout = computation_layout->result_layout(); in AddMandatoryConstraints() local
657 if (result_layout.LayoutIsSet()) { in AddMandatoryConstraints()
658 TF_RETURN_IF_ERROR(constraints->SetResultLayout(result_layout.shape())); in AddMandatoryConstraints()
[all …]
|
| D | local_service.cc | 104 if (build_options.result_layout() != nullptr) { in CreateExecutionOptions()
106 build_options.result_layout()->ToProto(); in CreateExecutionOptions()
163 if (build_options.result_layout() != nullptr) { in CompileExecutables()
164 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 | 97 instance.result_layout->ToProto(); in CompileAheadOfTime()
|
| D | hlo_verifier.cc | 1138 const ShapeLayout& result_layout = layout.result_layout(); in VerifyEntryComputationLayout() local
1141 ShapeUtil::ValidateShapeWithOptionalLayout(result_layout.shape())); in VerifyEntryComputationLayout()
1144 result_layout.shape())) { in VerifyEntryComputationLayout()
1149 ShapeUtil::HumanString(result_layout.shape())); in VerifyEntryComputationLayout()
1710 const Layout& result_layout = result_shape.layout(); in Postprocess() local
1716 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 | 322 module_config.entry_computation_layout().result_layout().shape(); in CreateFromProto()
|
| /external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/client/ |
| D | tpu_client.py | 96 if compile_options.result_layout:
97 options.result_layout = compile_options.result_layout
|
| D | tpu_client.cc | 817 ::xla::Shape result_layout; in Compile()
818 if (options.result_layout()) { in Compile()
819 result_layout = *options.result_layout(); in Compile()
830 result_layout = ::xla::Shape(program_shape_proto.result()); in Compile()
832 VLOG(1) << "Got result shape: " << result_layout.DebugString(); in Compile()
836 std::move(client), std::move(result_layout)); in Compile()
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | local_client.cc | 1084 Shape result_layout; in Compile() local
1085 if (options.result_layout()) { in Compile()
1086 result_layout = *options.result_layout(); in Compile()
1090 result_layout = program_shape.result(); in Compile()
1091 LayoutUtil::ClearLayout(&result_layout); in Compile()
1093 TF_RETURN_IF_ERROR(assign_layouts(&result_layout)); in Compile()
1094 options.set_result_layout(result_layout); in Compile()
|
| D | xla_client.py | 145 if compile_options.result_layout:
146 options.result_layout = compile_options.result_layout
525 self.result_layout = None
|
| D | xla.cc | 761 return options.result_layout() in PYBIND11_MODULE()
762 ? absl::optional<Shape>(*options.result_layout()) in PYBIND11_MODULE()
|
| /external/tensorflow/tensorflow/compiler/aot/ |
| D | compile.cc | 78 instance.result_layout = &result_shape; in CompileXla()
|