| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| cpu_layout_assignment_test.cc | 26 #include "tensorflow/compiler/xla/service/computation_layout.h"
72 ComputationLayout computation_layout(computation->ComputeProgramShape());
73 *computation_layout.mutable_parameter_layout(0) =
75 *computation_layout.mutable_result_layout() =
77 AssignLayouts(module.get(), &computation_layout);
113 ComputationLayout computation_layout(computation->ComputeProgramShape());
114 *computation_layout.mutable_parameter_layout(0) =
116 *computation_layout.mutable_result_layout() =
118 AssignLayouts(module.get(), &computation_layout);
157 ComputationLayout computation_layout(computation->ComputeProgramShape())
[all...] |
| cpu_layout_assignment.h | 19 #include "tensorflow/compiler/xla/service/computation_layout.h"
|
| BUILD | 199 "//tensorflow/compiler/xla/service:computation_layout",
665 "//tensorflow/compiler/xla/service:computation_layout",
685 "//tensorflow/compiler/xla/service:computation_layout",
|
| cpu_executable.cc | 27 #include "tensorflow/compiler/xla/service/computation_layout.h"
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| layout_assignment_test.cc | 26 #include "tensorflow/compiler/xla/service/computation_layout.h"
82 ComputationLayout computation_layout(computation->ComputeProgramShape());
83 *computation_layout.mutable_parameter_layout(0) = shape_layout;
84 *computation_layout.mutable_parameter_layout(1) = shape_layout;
85 *computation_layout.mutable_result_layout() = shape_layout;
86 AssignLayouts(module.get(), &computation_layout);
117 ComputationLayout computation_layout(computation->ComputeProgramShape());
118 *computation_layout.mutable_parameter_layout(0) = col_major;
119 *computation_layout.mutable_parameter_layout(1) = row_major;
120 *computation_layout.mutable_result_layout() = col_major
[all...] |
| computation_layout.cc | 16 #include "tensorflow/compiler/xla/service/computation_layout.h"
|
| layout_assignment.cc | 32 #include "tensorflow/compiler/xla/service/computation_layout.h"
456 const ComputationLayout& computation_layout,
483 &computation_layout.parameter_layout(instruction->parameter_number())
652 computation_layout.result_layout().shape());
661 const ComputationLayout& computation_layout) {
665 TF_RET_CHECK(computation_layout.parameter_layout(i).MatchesLayoutInShape(
669 computation_layout.result_layout().MatchesLayoutInShape(call->shape()));
753 const ComputationLayout& computation_layout)
[all...] |
| compile_only_service.cc | 24 #include "tensorflow/compiler/xla/service/computation_layout.h"
|
| hlo_module_config.h | 21 #include "tensorflow/compiler/xla/service/computation_layout.h"
|
| layout_assignment.h | 28 #include "tensorflow/compiler/xla/service/computation_layout.h"
358 const ComputationLayout& computation_layout,
374 Status RunOnComputation(const ComputationLayout& computation_layout,
|
| local_service.cc | 26 #include "tensorflow/compiler/xla/service/computation_layout.h"
|
| BUILD | 526 ":computation_layout",
568 ":computation_layout",
601 ":computation_layout",
675 ":computation_layout",
[all...] |
| executable.h | 23 #include "tensorflow/compiler/xla/service/computation_layout.h"
|
| service.cc | 28 #include "tensorflow/compiler/xla/service/computation_layout.h"
270 auto* computation_layout = config->mutable_entry_computation_layout(); local
290 computation_layout->mutable_parameter_layout(i)->CopyLayoutFromShape(
300 computation_layout->mutable_result_layout()->CopyLayoutFromShape(
303 computation_layout->mutable_result_layout()->Clear();
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| gpu_layout_assignment_test.cc | 19 #include "tensorflow/compiler/xla/service/computation_layout.h"
63 ComputationLayout computation_layout(
65 *computation_layout.mutable_parameter_layout(0) =
67 *computation_layout.mutable_parameter_layout(1) =
69 *computation_layout.mutable_result_layout() =
72 GpuLayoutAssignment layout_assignment(&computation_layout);
148 ComputationLayout computation_layout(computation->ComputeProgramShape());
151 *computation_layout.mutable_parameter_layout(0) =
156 *computation_layout.mutable_result_layout() = ShapeLayout(result_shape);
159 GpuLayoutAssignment layout_assignment(&computation_layout);
[all...] |
| gpu_layout_assignment.h | 19 #include "tensorflow/compiler/xla/service/computation_layout.h"
|
| BUILD | 581 "//tensorflow/compiler/xla/service:computation_layout",
597 "//tensorflow/compiler/xla/service:computation_layout",
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| local_client.cc | 56 const ComputationLayout& computation_layout = local
60 if (arguments.size() != computation_layout.parameter_count()) {
63 computation_layout.parameter_count(), arguments.size());
66 if (!computation_layout.parameter_layout(i).MatchesLayoutInShape(
73 ShapeUtil::HumanString(computation_layout.parameter_layout(i).shape())
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| hlo_test_base.h | 24 #include "tensorflow/compiler/xla/service/computation_layout.h"
|
| BUILD | 117 "//tensorflow/compiler/xla/service:computation_layout",
[all...] |
