Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Ddot_handler.cc287 auto output_slice_dim = lhs_concat_dim != -1 in GenNewConvDNums() local
290 if (output_slice_dim <= output_batch_dimension) { in GenNewConvDNums()
293 if (output_slice_dim <= output_feature_dimension) { in GenNewConvDNums()
297 if (output_slice_dim <= output_spatial_dimensions[i]) { in GenNewConvDNums()
301 output_spatial_dimensions.push_back(output_slice_dim); in GenNewConvDNums()