Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dspace_to_batch_converter.cc2156 int64 kernel_space_dim = original_conv_dims.kernel_spatial_dimensions( in PropagateOnBackpropFilterConv() local
2160 kernel_old->shape().dimensions(kernel_space_dim); in PropagateOnBackpropFilterConv()
2192 DimLookUp(permute_dims_kernel, kernel_space_dim); in PropagateOnBackpropFilterConv()
2240 kernel_old->shape().dimensions(kernel_space_dim); in PropagateOnBackpropFilterConv()
2247 auto retval, SplitSpace(kernel_old, tmp_dim_numbers, kernel_space_dim, in PropagateOnBackpropFilterConv()
2393 kernel_space_dim)); in PropagateOnBackpropFilterConv()