Home
last modified time | relevance | path

Searched refs:GpuLayoutAssignment (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_layout_assignment.h30 class GpuLayoutAssignment : public LayoutAssignment {
32 explicit GpuLayoutAssignment(ComputationLayout* entry_computation_layout, in GpuLayoutAssignment() function
39 ~GpuLayoutAssignment() override {} in ~GpuLayoutAssignment()
Dgpu_layout_assignment_test.cc79 GpuLayoutAssignment layout_assignment( in TEST_F()
168 GpuLayoutAssignment layout_assignment( in TEST_F()
239 GpuLayoutAssignment layout_assignment( in TEST_F()
321 GpuLayoutAssignment layout_assignment( in TEST_F()
358 GpuLayoutAssignment layout_assignment( in TEST_F()
396 GpuLayoutAssignment layout_assignment( in TEST_F()
423 GpuLayoutAssignment layout_assignment( in TEST_F()
Dgpu_layout_assignment.cc121 Status GpuLayoutAssignment::AddBackendConstraintsToDnnConvCustomCall( in AddBackendConstraintsToDnnConvCustomCall()
195 Status GpuLayoutAssignment::AddBackendConstraints( in AddBackendConstraints()
307 Status GpuLayoutAssignment::PropagateOperandConstraint( in PropagateOperandConstraint()
360 Status GpuLayoutAssignment::PropagateBufferConstraint( in PropagateBufferConstraint()
Dgpu_compiler.cc299 pipeline.AddPass<GpuLayoutAssignment>( in OptimizeHloModule()