Searched defs:computation_layout (Results 1 – 9 of 9) sorted by relevance
111 ComputationLayout computation_layout(computation->ComputeProgramShape()); in TEST_F() local146 ComputationLayout computation_layout(computation->ComputeProgramShape()); in TEST_F() local193 ComputationLayout computation_layout(computation->ComputeProgramShape()); in TEST_F() local234 ComputationLayout computation_layout( in TEST_F() local272 ComputationLayout computation_layout( in TEST_F() local307 ComputationLayout computation_layout( in TEST_F() local371 ComputationLayout computation_layout(computation->ComputeProgramShape()); in TEST_F() local412 ComputationLayout computation_layout(computation->ComputeProgramShape()); in TEST_F() local450 ComputationLayout computation_layout(computation->ComputeProgramShape()); in TEST_F() local492 ComputationLayout computation_layout(computation->ComputeProgramShape()); in TEST_F() local[all …]
50 ComputationLayout* computation_layout = in CreateModuleConfig() local
445 const ComputationLayout* computation_layout, in AddMandatoryConstraints()675 const ComputationLayout& computation_layout) { in CheckCallLayout()764 const ComputationLayout& computation_layout) { in CheckParameterLayout()1808 ComputationLayout computation_layout(computation->ComputeProgramShape(), in CalculateComputationLayout() local1842 ComputationLayout* computation_layout, HloComputation* computation, in RunOnComputation()2047 HloComputation* computation, ComputationLayout* computation_layout) { in PropagateComputationLayouts()2183 ComputationLayout* computation_layout = in Run() local
70 ComputationLayout computation_layout( in TEST_F() local157 ComputationLayout computation_layout(computation->ComputeProgramShape()); in TEST_F() local226 ComputationLayout computation_layout(computation->ComputeProgramShape()); in TEST_F() local307 ComputationLayout computation_layout( in TEST_F() local355 ComputationLayout computation_layout( in TEST_F() local393 ComputationLayout computation_layout( in TEST_F() local420 ComputationLayout computation_layout( in TEST_F() local
79 ComputationLayout computation_layout(computation->ComputeProgramShape()); in TEST_F() local120 ComputationLayout computation_layout(computation->ComputeProgramShape()); in TEST_F() local164 ComputationLayout computation_layout(computation->ComputeProgramShape()); in TEST_F() local198 ComputationLayout computation_layout(computation->ComputeProgramShape()); in TEST_F() local238 ComputationLayout computation_layout(computation->ComputeProgramShape()); in TEST_F() local313 ComputationLayout computation_layout(computation->ComputeProgramShape()); in RunDotOutputFusion() local458 ComputationLayout computation_layout(computation->ComputeProgramShape()); in TEST_F() local
78 const xla::ComputationLayout& computation_layout) { in GetDynamicInputInfo()92 const xla::ComputationLayout& computation_layout = in GetInputTuples() local302 const xla::ComputationLayout& computation_layout = in RunExecutable() local
121 const ComputationLayout& computation_layout = in RunHelper() local
452 xla::ComputationLayout computation_layout( in TPUExecute() local
65 ComputationLayout* computation_layout = in CreateModuleConfig() local