Searched refs:assign_layouts (Results 1 – 1 of 1) sorted by relevance
170 auto assign_layouts = [&choose_compact_layout_for_shape_function]( in DetermineArgumentLayoutsFromCompileOptions() local194 TF_RETURN_IF_ERROR(assign_layouts(sharded_shapes.first[i], layout)); in DetermineArgumentLayoutsFromCompileOptions()204 TF_RETURN_IF_ERROR(assign_layouts(sharded_shapes.second, &result_layout)); in DetermineArgumentLayoutsFromCompileOptions()