| /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 | hlo_module_util.cc | 50 ComputationLayout* computation_layout = in CreateModuleConfig() local
70 computation_layout->mutable_parameter_layout(i)->CopyLayoutFromShape( in CreateModuleConfig()
80 computation_layout->mutable_result_layout()->CopyLayoutFromShape( in CreateModuleConfig()
84 computation_layout->mutable_result_layout()->SetToDefaultLayout(); in CreateModuleConfig()
|
| D | layout_assignment.cc | 445 const ComputationLayout* computation_layout, in AddMandatoryConstraints() argument
472 if (computation_layout != nullptr) { in AddMandatoryConstraints()
474 computation_layout->parameter_layout( in AddMandatoryConstraints()
656 } else if (computation_layout != nullptr) { in AddMandatoryConstraints()
657 const ShapeLayout& result_layout = computation_layout->result_layout(); in AddMandatoryConstraints()
675 const ComputationLayout& computation_layout) { in CheckCallLayout() argument
679 TF_RET_CHECK(computation_layout.parameter_layout(i).MatchesLayoutInShape( in CheckCallLayout()
682 TF_RET_CHECK(computation_layout.result_layout().MatchesLayoutInShape( in CheckCallLayout()
764 const ComputationLayout& computation_layout) { in CheckParameterLayout() argument
766 computation_layout.parameter_layout(parameter->parameter_number()); in CheckParameterLayout()
[all …]
|
| D | layout_assignment.h | 388 Status AddMandatoryConstraints(const ComputationLayout* computation_layout,
407 Status RunOnComputation(ComputationLayout* computation_layout,
448 ComputationLayout* computation_layout);
|
| D | BUILD | 941 ":computation_layout",
989 ":computation_layout",
1025 ":computation_layout",
1130 ":computation_layout",
3531 ":computation_layout",
3963 ":computation_layout",
4338 ":computation_layout",
4351 name = "computation_layout",
4352 srcs = ["computation_layout.cc"],
4353 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 | 336 "//tensorflow/compiler/xla/service:computation_layout",
950 "//tensorflow/compiler/xla/service:computation_layout",
972 "//tensorflow/compiler/xla/service:computation_layout",
1070 "//tensorflow/compiler/xla/service:computation_layout",
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_layout_assignment_test.cc | 70 ComputationLayout computation_layout( in TEST_F() local
72 *computation_layout.mutable_parameter_layout(0) = in TEST_F()
74 *computation_layout.mutable_parameter_layout(1) = in TEST_F()
76 *computation_layout.mutable_result_layout() = in TEST_F()
80 &computation_layout, LayoutAssignment::InstructionCanChangeLayout, in TEST_F()
157 ComputationLayout computation_layout(computation->ComputeProgramShape()); in TEST_F() local
160 *computation_layout.mutable_parameter_layout(0) = in TEST_F()
165 *computation_layout.mutable_result_layout() = ShapeLayout(result_shape); in TEST_F()
169 &computation_layout, LayoutAssignment::InstructionCanChangeLayout, in TEST_F()
226 ComputationLayout computation_layout(computation->ComputeProgramShape()); in TEST_F() local
[all …]
|
| D | BUILD | 1047 "//tensorflow/compiler/xla/service:computation_layout",
1514 "//tensorflow/compiler/xla/service:computation_layout",
1534 "//tensorflow/compiler/xla/service:computation_layout",
|
| /external/tensorflow/tensorflow/compiler/xrt/kernels/ |
| D | xrt_execute_op.cc | 78 const xla::ComputationLayout& computation_layout) { in GetDynamicInputInfo() argument
80 input_is_dynamic.reserve(computation_layout.parameter_count()); in GetDynamicInputInfo()
81 for (int64 i = 0; i < computation_layout.parameter_count(); ++i) { in GetDynamicInputInfo()
83 !computation_layout.parameter_shape(i).is_static()); in GetDynamicInputInfo()
92 const xla::ComputationLayout& computation_layout = in GetInputTuples() local
96 input_coords, working_set, backend, computation_layout.parameter_count(), in GetInputTuples()
97 [&](int64 i) { return computation_layout.parameter_shape(i); }, in GetInputTuples()
302 const xla::ComputationLayout& computation_layout = in RunExecutable() local
304 std::vector<bool> input_is_dynamic = GetDynamicInputInfo(computation_layout); in RunExecutable()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | local_client.cc | 121 const ComputationLayout& computation_layout = in RunHelper() local
126 if (argument_shapes_size != computation_layout.parameter_count()) { in RunHelper()
129 computation_layout.parameter_count(), argument_shapes.size()); in RunHelper()
132 if (!computation_layout.parameter_layout(i).MatchesLayoutInShape( in RunHelper()
141 computation_layout.parameter_layout(i).shape()), in RunHelper()
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | tpu_compile_op_support.cc | 65 ComputationLayout* computation_layout = in CreateModuleConfig() local
84 computation_layout->mutable_parameter_layout(i)->CopyLayoutFromShape( in CreateModuleConfig()
92 computation_layout->mutable_result_layout()->CopyLayoutFromShape( in CreateModuleConfig()
96 computation_layout->mutable_result_layout()->SetToDefaultLayout(); in CreateModuleConfig()
|
| D | BUILD | 194 "//tensorflow/compiler/xla/service:computation_layout",
|
| /external/tensorflow/tensorflow/core/tpu/ |
| D | tpu_execute.cc | 452 xla::ComputationLayout computation_layout( in TPUExecute() local
456 computation_layout.add_parameter_layout(xla::ShapeLayout(shape)); in TPUExecute()
461 xla::HloModuleConfig(std::move(computation_layout))); in TPUExecute()
|
| D | BUILD | 230 "//tensorflow/compiler/xla/service:computation_layout",
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | BUILD | 166 "//tensorflow/compiler/xla/service:computation_layout",
|