Searched refs:count_of_dimensions_to_convert (Results 1 – 2 of 2) sorted by relevance
37 int64_t count_of_dimensions_to_convert = 1; member
244 const int64_t dim_count = ctrl_.count_of_dimensions_to_convert; in GetFirstChosenSpatialDim()252 const int64_t dim_count = ctrl_.count_of_dimensions_to_convert; in GetChosenSpatialDims()360 std::vector<int64_t> old_dims(ctrl_.count_of_dimensions_to_convert), in GetSpatialDimsToSplit()361 new_dims(ctrl_.count_of_dimensions_to_convert); in GetSpatialDimsToSplit()365 for (int i = 1; i < ctrl_.count_of_dimensions_to_convert; ++i) { in GetSpatialDimsToSplit()2395 ctrl_.count_of_dimensions_to_convert); in BatchToSpace()2407 ctrl_.count_of_dimensions_to_convert); in BatchToSpace()2707 for (int i = 0; i < ctrl_.count_of_dimensions_to_convert; ++i) { in PropagateOnConv()3181 ctrl_.count_of_dimensions_to_convert); in PropagateOnBackpropFilterConv()3183 for (int64_t i = 0; i < ctrl_.count_of_dimensions_to_convert; ++i) { in PropagateOnBackpropFilterConv()[all …]