| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | layout_assignment_test.cc | 111 ComputationLayout computation_layout(computation->ComputeProgramShape()); in TEST_F() local
112 *computation_layout.mutable_parameter_layout(0) = shape_layout; in TEST_F()
113 *computation_layout.mutable_parameter_layout(1) = shape_layout; in TEST_F()
114 *computation_layout.mutable_result_layout() = shape_layout; in TEST_F()
115 AssignLayouts(m.get(), &computation_layout); in TEST_F()
146 ComputationLayout computation_layout(computation->ComputeProgramShape()); in TEST_F() local
147 *computation_layout.mutable_parameter_layout(0) = col_major; in TEST_F()
148 *computation_layout.mutable_parameter_layout(1) = row_major; in TEST_F()
149 *computation_layout.mutable_result_layout() = col_major; in TEST_F()
151 AssignLayouts(m.get(), &computation_layout); in TEST_F()
[all …]
|
| D | layout_assignment.cc | 429 const ComputationLayout* computation_layout, in AddMandatoryConstraints() argument
456 if (computation_layout != nullptr) { in AddMandatoryConstraints()
458 computation_layout->parameter_layout( in AddMandatoryConstraints()
643 if (computation_layout != nullptr) { in AddMandatoryConstraints()
644 const ShapeLayout& result_layout = computation_layout->result_layout(); in AddMandatoryConstraints()
658 const ComputationLayout& computation_layout) { in CheckCallLayout() argument
662 TF_RET_CHECK(computation_layout.parameter_layout(i).MatchesLayoutInShape( in CheckCallLayout()
666 computation_layout.result_layout().MatchesLayoutInShape(call->shape())); in CheckCallLayout()
745 const ComputationLayout& computation_layout) { in CheckParameterLayout() argument
747 computation_layout.parameter_layout(parameter->parameter_number()); in CheckParameterLayout()
[all …]
|
| D | layout_assignment.h | 379 Status AddMandatoryConstraints(const ComputationLayout* computation_layout,
398 Status RunOnComputation(ComputationLayout* computation_layout,
440 ComputationLayout* computation_layout);
|
| D | service.cc | 269 ComputationLayout* computation_layout = in CreateModuleConfig() local
288 computation_layout->mutable_parameter_layout(i)->CopyLayoutFromShape( in CreateModuleConfig()
298 computation_layout->mutable_result_layout()->CopyLayoutFromShape( in CreateModuleConfig()
302 computation_layout->mutable_result_layout()->SetToDefaultLayout(); in CreateModuleConfig()
|
| D | BUILD | 721 ":computation_layout",
768 ":computation_layout",
803 ":computation_layout",
902 ":computation_layout",
2600 ":computation_layout",
2841 ":computation_layout",
3229 ":computation_layout",
3242 name = "computation_layout",
3243 srcs = ["computation_layout.cc"],
3244 hdrs = ["computation_layout.h"],
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_layout_assignment_test.cc | 79 ComputationLayout computation_layout(computation->ComputeProgramShape()); in TEST_F() local
80 *computation_layout.mutable_parameter_layout(0) = in TEST_F()
82 *computation_layout.mutable_result_layout() = in TEST_F()
84 AssignLayouts(module.get(), &computation_layout); in TEST_F()
120 ComputationLayout computation_layout(computation->ComputeProgramShape()); in TEST_F() local
121 *computation_layout.mutable_parameter_layout(0) = in TEST_F()
123 *computation_layout.mutable_result_layout() = in TEST_F()
125 AssignLayouts(module.get(), &computation_layout); in TEST_F()
164 ComputationLayout computation_layout(computation->ComputeProgramShape()); in TEST_F() local
165 *computation_layout.mutable_parameter_layout(0) = in TEST_F()
[all …]
|
| D | BUILD | 239 "//tensorflow/compiler/xla/service:computation_layout",
797 "//tensorflow/compiler/xla/service:computation_layout",
819 "//tensorflow/compiler/xla/service:computation_layout",
914 "//tensorflow/compiler/xla/service:computation_layout",
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_layout_assignment_test.cc | 68 ComputationLayout computation_layout( in TEST_F() local
70 *computation_layout.mutable_parameter_layout(0) = in TEST_F()
72 *computation_layout.mutable_parameter_layout(1) = in TEST_F()
74 *computation_layout.mutable_result_layout() = in TEST_F()
78 &computation_layout, LayoutAssignment::InstructionCanChangeLayout, in TEST_F()
155 ComputationLayout computation_layout(computation->ComputeProgramShape()); in TEST_F() local
158 *computation_layout.mutable_parameter_layout(0) = in TEST_F()
163 *computation_layout.mutable_result_layout() = ShapeLayout(result_shape); in TEST_F()
167 &computation_layout, LayoutAssignment::InstructionCanChangeLayout, in TEST_F()
224 ComputationLayout computation_layout(computation->ComputeProgramShape()); in TEST_F() local
[all …]
|
| D | BUILD | 683 "//tensorflow/compiler/xla/service:computation_layout",
948 "//tensorflow/compiler/xla/service:computation_layout",
966 "//tensorflow/compiler/xla/service:computation_layout",
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | local_client.cc | 57 const ComputationLayout& computation_layout = in ValidateExecutionOptions() local
61 if (arguments.size() != computation_layout.parameter_count()) { in ValidateExecutionOptions()
64 computation_layout.parameter_count(), arguments.size()); in ValidateExecutionOptions()
67 if (!computation_layout.parameter_layout(i).MatchesLayoutInShape( in ValidateExecutionOptions()
76 computation_layout.parameter_layout(i).shape()), in ValidateExecutionOptions()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | BUILD | 125 "//tensorflow/compiler/xla/service:computation_layout",
|