Searched refs:patch_depths (Results 1 – 2 of 2) sorted by relevance
604 const int64 patch_depths = filter.dim_size(2); in operator ()() local609 bool is_grouped_convolution = patch_depths != in_depths; in operator ()()616 const uint64 k = patch_depths; in operator ()()644 const uint64 k = patch_rows * patch_cols * patch_depths; in operator ()()778 .set_input_feature_map_count(patch_depths) in operator ()()787 .set_group_count(in_depths / patch_depths); in operator ()()837 patch_depths}}, // filter_depths in operator ()()
588 const int64 patch_depths = filter.dim_size(2);687 .set_input_feature_map_count(patch_depths)702 .set_group_count(in_depths / patch_depths);761 patch_depths}}, // filter_depths