Searched refs:patch_depths (Results 1 – 2 of 2) sorted by relevance
672 const int64 patch_depths = filter.dim_size(2); in operator ()() local677 bool is_grouped_convolution = patch_depths != in_depths; in operator ()()684 const uint64 k = patch_depths; in operator ()()712 const uint64 k = patch_rows * patch_cols * patch_depths; in operator ()()894 .set_input_feature_map_count(patch_depths) in operator ()()904 .set_group_count(in_depths / patch_depths); in operator ()()975 patch_depths}}, // filter_depths in operator ()()
467 const int64 patch_depths = filter.dim_size(2);581 .set_input_feature_map_count(patch_depths)596 .set_group_count(in_depths / patch_depths);654 patch_depths}}, // filter_depths