Searched refs:mutable_result_layout (Results 1 – 9 of 9) 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()732 computation_layout.mutable_result_layout()->ResetLayout( in TEST_F()[all …]
68 ShapeLayout* mutable_result_layout() { return &result_layout_; } in mutable_result_layout() function
298 computation_layout->mutable_result_layout()->CopyLayoutFromShape( in CreateModuleConfig()302 computation_layout->mutable_result_layout()->SetToDefaultLayout(); in CreateModuleConfig()
387 TF_RETURN_IF_ERROR(entry_layout->mutable_result_layout()->CopyLayoutFromShape( in CreateModuleConfigFromProto()
629 *branch_computation_layout.mutable_result_layout() = in AddMandatoryConstraints()1906 ShapeLayout* result_layout = computation_layout->mutable_result_layout(); in PropagateComputationLayouts()
541 ->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()311 *computation_layout.mutable_result_layout() = in RunDotOutputFusion()
277 ->mutable_result_layout() in ForceResultLayout()284 ->mutable_result_layout() in ForceResultLayout()292 ->mutable_result_layout() in ForceClearResultLayout()
74 *computation_layout.mutable_result_layout() = in TEST_F()163 *computation_layout.mutable_result_layout() = ShapeLayout(result_shape); in TEST_F()232 *computation_layout.mutable_result_layout() = in TEST_F()314 *computation_layout.mutable_result_layout() = in TEST_F()