Home
last modified time | relevance | path

Searched refs:new_dims_mapping (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Ddot_handler.cc2250 DotConvDimsMapping new_dims_mapping; in ConvertDimsMappingWithFeatureGroupCount() local
2251 new_dims_mapping.batch_dims = dims_mapping.batch_dims; in ConvertDimsMappingWithFeatureGroupCount()
2252 new_dims_mapping.conv_spatial_dims = dims_mapping.conv_spatial_dims; in ConvertDimsMappingWithFeatureGroupCount()
2254 new_dims_mapping.batch_dims.emplace_back(); in ConvertDimsMappingWithFeatureGroupCount()
2255 new_dims_mapping.batch_dims.back().lhs = dnums.input_feature_dimension(); in ConvertDimsMappingWithFeatureGroupCount()
2256 new_dims_mapping.batch_dims.back().rhs = in ConvertDimsMappingWithFeatureGroupCount()
2258 new_dims_mapping.batch_dims.back().output = dnums.output_feature_dimension(); in ConvertDimsMappingWithFeatureGroupCount()
2259 new_dims_mapping.batch_dims.back().spatial = -1; in ConvertDimsMappingWithFeatureGroupCount()
2261 new_dims_mapping.lhs_non_contracting_dims.emplace_back(); in ConvertDimsMappingWithFeatureGroupCount()
2262 new_dims_mapping.lhs_non_contracting_dims.back().lhs = in ConvertDimsMappingWithFeatureGroupCount()
[all …]