Searched refs:mutable_result_layout (Results 1 – 11 of 11) sorted by relevance
114 *computation_layout.mutable_result_layout() = shape_layout; in TEST_F()149 *computation_layout.mutable_result_layout() = col_major; in TEST_F()194 *computation_layout.mutable_result_layout() = shape_layout; in TEST_F()276 TF_CHECK_OK(computation_layout.mutable_result_layout()->CopyLayoutFromShape( in TEST_F()314 TF_CHECK_OK(computation_layout.mutable_result_layout()->CopyLayoutFromShape( in TEST_F()374 *computation_layout.mutable_result_layout() = ShapeLayout(bshape_with_layout); in TEST_F()415 *computation_layout.mutable_result_layout() = ShapeLayout(bshape_with_layout); in TEST_F()453 *computation_layout.mutable_result_layout() = in TEST_F()504 *computation_layout.mutable_result_layout() = in TEST_F()729 computation_layout.mutable_result_layout()->ResetLayout( in TEST_F()[all …]
80 computation_layout->mutable_result_layout()->CopyLayoutFromShape( in CreateModuleConfig()84 computation_layout->mutable_result_layout()->SetToDefaultLayout(); in CreateModuleConfig()
68 ShapeLayout* mutable_result_layout() { return &result_layout_; } in mutable_result_layout() function
477 TF_RETURN_IF_ERROR(entry_layout->mutable_result_layout()->CopyLayoutFromShape( in CreateModuleConfigFromShape()
385 ->mutable_result_layout() in PadAndExecute()
1943 *FindOrDie(computation_layouts_, computation).mutable_result_layout() = in RunOnComputation()2082 ShapeLayout* result_layout = computation_layout->mutable_result_layout(); in PropagateComputationLayouts()
864 ->mutable_result_layout() in ParseComputations()
82 *computation_layout.mutable_result_layout() = in TEST_F()123 *computation_layout.mutable_result_layout() = in TEST_F()169 *computation_layout.mutable_result_layout() = in TEST_F()201 *computation_layout.mutable_result_layout() = in TEST_F()241 *computation_layout.mutable_result_layout() = in TEST_F()318 *computation_layout.mutable_result_layout() = in RunDotOutputFusion()463 *computation_layout.mutable_result_layout() = in TEST_F()
278 ->mutable_result_layout() in ForceResultLayout()285 ->mutable_result_layout() in ForceResultLayout()293 ->mutable_result_layout() in ForceClearResultLayout()
76 *computation_layout.mutable_result_layout() = in TEST_F()165 *computation_layout.mutable_result_layout() = ShapeLayout(result_shape); in TEST_F()234 *computation_layout.mutable_result_layout() = in TEST_F()316 *computation_layout.mutable_result_layout() = in TEST_F()
92 computation_layout->mutable_result_layout()->CopyLayoutFromShape( in CreateModuleConfig()96 computation_layout->mutable_result_layout()->SetToDefaultLayout(); in CreateModuleConfig()