Searched defs:computation_layout (Results 1 – 6 of 6) 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 …]
429 const ComputationLayout* computation_layout, in AddMandatoryConstraints()658 const ComputationLayout& computation_layout) { in CheckCallLayout()745 const ComputationLayout& computation_layout) { in CheckParameterLayout()1667 ComputationLayout computation_layout(computation->ComputeProgramShape(), in CalculateComputationLayout() local1700 ComputationLayout* computation_layout, in RunOnComputation()1890 HloComputation* computation, ComputationLayout* computation_layout) { in PropagateComputationLayouts()1961 ComputationLayout* computation_layout = in Run() local
269 ComputationLayout* computation_layout = in CreateModuleConfig() local
68 ComputationLayout computation_layout( in TEST_F() local155 ComputationLayout computation_layout(computation->ComputeProgramShape()); in TEST_F() local224 ComputationLayout computation_layout(computation->ComputeProgramShape()); in TEST_F() local305 ComputationLayout computation_layout( in TEST_F() local353 ComputationLayout computation_layout( in TEST_F() local391 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() local306 ComputationLayout computation_layout(computation->ComputeProgramShape()); in RunDotOutputFusion() local
57 const ComputationLayout& computation_layout = in ValidateExecutionOptions() local