HomeSort by relevance Sort by last modified time
    Searched refs:computation_layout (Results 1 - 7 of 7) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/service/
layout_assignment_test.cc 27 #include "tensorflow/compiler/xla/service/computation_layout.h"
111 ComputationLayout computation_layout(computation->ComputeProgramShape());
112 *computation_layout.mutable_parameter_layout(0) = shape_layout;
113 *computation_layout.mutable_parameter_layout(1) = shape_layout;
114 *computation_layout.mutable_result_layout() = shape_layout;
115 AssignLayouts(m.get(), &computation_layout);
146 ComputationLayout computation_layout(computation->ComputeProgramShape());
147 *computation_layout.mutable_parameter_layout(0) = col_major;
148 *computation_layout.mutable_parameter_layout(1) = row_major;
149 *computation_layout.mutable_result_layout() = col_major
1151 ComputationLayout computation_layout = m->entry_computation_layout(); local
1167 ComputationLayout computation_layout = m->entry_computation_layout(); local
1194 ComputationLayout computation_layout = m->entry_computation_layout(); local
1226 ComputationLayout computation_layout = m->entry_computation_layout(); local
1253 ComputationLayout computation_layout = m->entry_computation_layout(); local
1288 ComputationLayout computation_layout = m->entry_computation_layout(); local
    [all...]
layout_assignment.cc 36 #include "tensorflow/compiler/xla/service/computation_layout.h"
429 const ComputationLayout* computation_layout,
456 if (computation_layout != nullptr) {
458 computation_layout->parameter_layout(
643 if (computation_layout != nullptr) {
644 const ShapeLayout& result_layout = computation_layout->result_layout();
658 const ComputationLayout& computation_layout) {
662 TF_RET_CHECK(computation_layout.parameter_layout(i).MatchesLayoutInShape(
666 computation_layout.result_layout().MatchesLayoutInShape(call->shape()));
745 const ComputationLayout& computation_layout) {
    [all...]
layout_assignment.h 30 #include "tensorflow/compiler/xla/service/computation_layout.h"
379 Status AddMandatoryConstraints(const ComputationLayout* computation_layout,
398 Status RunOnComputation(ComputationLayout* computation_layout,
440 ComputationLayout* computation_layout);
service.cc 30 #include "tensorflow/compiler/xla/service/computation_layout.h"
269 ComputationLayout* computation_layout = local
288 computation_layout->mutable_parameter_layout(i)->CopyLayoutFromShape(
298 computation_layout->mutable_result_layout()->CopyLayoutFromShape(
302 computation_layout->mutable_result_layout()->SetToDefaultLayout();
    [all...]
  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
gpu_layout_assignment_test.cc 20 #include "tensorflow/compiler/xla/service/computation_layout.h"
68 ComputationLayout computation_layout(
70 *computation_layout.mutable_parameter_layout(0) =
72 *computation_layout.mutable_parameter_layout(1) =
74 *computation_layout.mutable_result_layout() =
78 &computation_layout, LayoutAssignment::InstructionCanChangeLayout,
155 ComputationLayout computation_layout(computation->ComputeProgramShape());
158 *computation_layout.mutable_parameter_layout(0) =
163 *computation_layout.mutable_result_layout() = ShapeLayout(result_shape);
167 &computation_layout, LayoutAssignment::InstructionCanChangeLayout
    [all...]
  /external/tensorflow/tensorflow/compiler/xla/service/cpu/
cpu_layout_assignment_test.cc 27 #include "tensorflow/compiler/xla/service/computation_layout.h"
79 ComputationLayout computation_layout(computation->ComputeProgramShape());
80 *computation_layout.mutable_parameter_layout(0) =
82 *computation_layout.mutable_result_layout() =
84 AssignLayouts(module.get(), &computation_layout);
120 ComputationLayout computation_layout(computation->ComputeProgramShape());
121 *computation_layout.mutable_parameter_layout(0) =
123 *computation_layout.mutable_result_layout() =
125 AssignLayouts(module.get(), &computation_layout);
164 ComputationLayout computation_layout(computation->ComputeProgramShape())
    [all...]
  /external/tensorflow/tensorflow/compiler/xla/client/
local_client.cc 57 const ComputationLayout& computation_layout = local
61 if (arguments.size() != computation_layout.parameter_count()) {
64 computation_layout.parameter_count(), arguments.size());
67 if (!computation_layout.parameter_layout(i).MatchesLayoutInShape(
76 computation_layout.parameter_layout(i).shape()),

Completed in 182 milliseconds