Home
last modified time | relevance | path

Searched refs:ComputationLayout (Results 1 – 15 of 15) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dcomputation_layout.cc26 ComputationLayout::ComputationLayout(const ProgramShape& program_shape, in ComputationLayout() function in xla::ComputationLayout
37 void ComputationLayout::SetToDefaultLayout() { in SetToDefaultLayout()
44 bool ComputationLayout::LayoutIsSet() const { in LayoutIsSet()
50 string ComputationLayout::ToString() const { in ToString()
59 ProgramShape ComputationLayout::ComputeProgramShape() const { in ComputeProgramShape()
Dlayout_assignment_test.cc54 void AssignLayouts(HloModule* m, ComputationLayout* entry_computation_layout, in AssignLayouts()
90 TEST_F(LayoutAssignmentTest, ComputationLayout) { in TEST_F() argument
111 ComputationLayout computation_layout(computation->ComputeProgramShape()); in TEST_F()
146 ComputationLayout computation_layout(computation->ComputeProgramShape()); in TEST_F()
193 ComputationLayout computation_layout(computation->ComputeProgramShape()); in TEST_F()
234 ComputationLayout computation_layout( in TEST_F()
272 ComputationLayout computation_layout( in TEST_F()
307 ComputationLayout computation_layout( in TEST_F()
371 ComputationLayout computation_layout(computation->ComputeProgramShape()); in TEST_F()
412 ComputationLayout computation_layout(computation->ComputeProgramShape()); in TEST_F()
[all …]
Dcomputation_layout.h33 class ComputationLayout {
36 explicit ComputationLayout(ShapeLayout result_layout) in ComputationLayout() function
42 explicit ComputationLayout(const ProgramShape& program_shape,
Dlayout_assignment.h302 ComputationLayout* entry_computation_layout,
379 Status AddMandatoryConstraints(const ComputationLayout* computation_layout,
398 Status RunOnComputation(ComputationLayout* computation_layout,
440 ComputationLayout* computation_layout);
443 ComputationLayout* entry_computation_layout_;
447 ComputationLayout saved_entry_computation_layout_;
505 std::map<HloComputation*, ComputationLayout> computation_layouts_;
Dhlo_module_config.h59 const ComputationLayout& entry_computation_layout() const { in entry_computation_layout()
66 ComputationLayout* mutable_entry_computation_layout() { in mutable_entry_computation_layout()
122 absl::optional<ComputationLayout> entry_computation_layout_;
Dhlo_module.h122 ComputationLayout compute_computation_layout() const { in compute_computation_layout()
123 return ComputationLayout(entry_computation()->ComputeProgramShape(), in compute_computation_layout()
127 ComputationLayout* mutable_entry_computation_layout() { in mutable_entry_computation_layout()
131 const ComputationLayout& entry_computation_layout() const { in entry_computation_layout()
Dhlo_module_config.cc34 ComputationLayout(program_shape, ignore_layouts)) {} in HloModuleConfig()
38 entry_computation_layout_ = ComputationLayout(program_shape); in SetDefaultComputationLayout()
Dlayout_assignment.cc429 const ComputationLayout* computation_layout, in AddMandatoryConstraints()
536 const ComputationLayout& called_computation_layout = in AddMandatoryConstraints()
554 ComputationLayout& body_layout = FindOrDie(computation_layouts_, body); in AddMandatoryConstraints()
555 ComputationLayout& condition_layout = in AddMandatoryConstraints()
604 ComputationLayout& best_branch_computation_layout = in AddMandatoryConstraints()
611 ComputationLayout& branch_computation_layout = in AddMandatoryConstraints()
658 const ComputationLayout& computation_layout) { in CheckCallLayout()
692 const ComputationLayout& condition_computation_layout, in CheckWhileLayout()
693 const ComputationLayout& body_computation_layout) { in CheckWhileLayout()
709 absl::Span<const ComputationLayout> branch_computation_layouts) { in CheckConditionalLayout()
[all …]
Dhlo_module.cc380 ComputationLayout* entry_layout = in CreateModuleConfigFromProto()
Dservice.cc269 ComputationLayout* computation_layout = in CreateModuleConfig()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_layout_assignment_test.cc68 ComputationLayout computation_layout( in TEST_F()
155 ComputationLayout computation_layout(computation->ComputeProgramShape()); in TEST_F()
224 ComputationLayout computation_layout(computation->ComputeProgramShape()); in TEST_F()
305 ComputationLayout computation_layout( in TEST_F()
353 ComputationLayout computation_layout( in TEST_F()
391 ComputationLayout computation_layout( in TEST_F()
Dgpu_layout_assignment.h32 explicit GpuLayoutAssignment(ComputationLayout* entry_computation_layout, in GpuLayoutAssignment()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_layout_assignment_test.cc52 ComputationLayout* entry_computation_layout) { in AssignLayouts()
79 ComputationLayout computation_layout(computation->ComputeProgramShape()); in TEST_F()
120 ComputationLayout computation_layout(computation->ComputeProgramShape()); in TEST_F()
164 ComputationLayout computation_layout(computation->ComputeProgramShape()); in TEST_F()
198 ComputationLayout computation_layout(computation->ComputeProgramShape()); in TEST_F()
238 ComputationLayout computation_layout(computation->ComputeProgramShape()); in TEST_F()
306 ComputationLayout computation_layout(computation->ComputeProgramShape()); in RunDotOutputFusion()
Dcpu_layout_assignment.h32 ComputationLayout* entry_computation_layout, in CpuLayoutAssignment()
/external/tensorflow/tensorflow/compiler/xla/client/
Dlocal_client.cc57 const ComputationLayout& computation_layout = in ValidateExecutionOptions()