Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dspace_to_batch_converter.cc2235 const int64 new_split_dim_size = in PropagateOnBackpropFilterConv() local
2238 CeilOfRatio(new_split_dim_size, rhs_dilation); in PropagateOnBackpropFilterConv()
2332 int64 new_split_dim_size = in PropagateOnBackpropFilterConv() local
2344 << " new_split_dim_size " << new_split_dim_size; in PropagateOnBackpropFilterConv()
2362 if (new_split_dim_size != expected_split_dim_size) { in PropagateOnBackpropFilterConv()
2363 CHECK_LT(new_split_dim_size, expected_split_dim_size); in PropagateOnBackpropFilterConv()
2364 new_split_dim_size = expected_split_dim_size; in PropagateOnBackpropFilterConv()
2369 spatial_dimension_to_split, new_split_dim_size)); in PropagateOnBackpropFilterConv()
2517 ->set_size(CeilOfRatio(new_split_dim_size, rhs_dilation)); in PropagateOnBackpropFilterConv()