Searched defs:new_space_dim (Results 1 – 1 of 1) sorted by relevance
1211 const int64_t new_space_dim = in CanPropagate() local1375 const int64_t new_space_dim = in CanPropagate() local1422 const int64_t new_space_dim = DimLookUp(permute_dims, old_space_dim); in PropagateOnBroadcast() local1696 const int64_t new_space_dim = DimLookUp(permute_dims, old_space_dim); in SupportedOpForPropagation() local1967 int64_t new_batch_dim, new_space_dim, new_feature_dim; in Propagate() local2018 const int64_t new_space_dim = new_spatial_dims[0]; in Propagate() local2977 const int64_t new_space_dim = new_spatial_dims[0]; in PropagateOnConstant() local3090 const int64_t new_space_dim = in PropagateOnBackpropFilterConv() local3530 const int64_t new_space_dim = in IsSpaceToBatchedSpaceSizeSuitable() local