HomeSort by relevance Sort by last modified time
    Searched refs:conv_dnums (Results 1 - 3 of 3) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
cudnn_convolution_rewriter.cc 79 const ConvolutionDimensionNumbers& conv_dnums = local
81 auto input_batch_dim = conv_dnums.input_batch_dimension();
82 auto input_feature_dim = conv_dnums.input_feature_dimension();
83 auto input_spatial_dims = conv_dnums.input_spatial_dimensions();
84 auto kernel_input_feature_dim = conv_dnums.kernel_input_feature_dimension();
85 auto kernel_output_feature_dim = conv_dnums.kernel_output_feature_dimension();
86 auto kernel_spatial_dims = conv_dnums.kernel_spatial_dimensions();
87 auto output_batch_dim = conv_dnums.output_batch_dimension();
88 auto output_feature_dim = conv_dnums.output_feature_dimension();
89 auto output_spatial_dims = conv_dnums.output_spatial_dimensions()
    [all...]
cudnn_convolution_rewriter_test.cc 256 ConvolutionDimensionNumbers conv_dnums; local
257 conv_dnums.set_input_batch_dimension(0);
258 conv_dnums.set_output_batch_dimension(0);
259 conv_dnums.set_input_feature_dimension(1);
260 conv_dnums.set_output_feature_dimension(1);
261 conv_dnums.add_input_spatial_dimensions(2);
262 conv_dnums.add_output_spatial_dimensions(2);
263 conv_dnums.add_input_spatial_dimensions(3);
264 conv_dnums.add_output_spatial_dimensions(3);
265 conv_dnums.set_kernel_input_feature_dimension(0)
    [all...]
pad_insertion.cc 43 const Window& conv_window, const ConvolutionDimensionNumbers& conv_dnums,
58 for (size_t i = 0; i < conv_dnums.input_spatial_dimensions().size(); ++i) {
59 int64 dim = conv_dnums.input_spatial_dimensions(i);
90 for (size_t i = 0; i < conv_dnums.input_spatial_dimensions().size(); ++i) {
91 int64 dim = conv_dnums.input_spatial_dimensions(i);
115 const ConvolutionDimensionNumbers& conv_dnums,
126 for (size_t i = 0; i < conv_dnums.kernel_spatial_dimensions().size(); ++i) {
127 int64 dim = conv_dnums.kernel_spatial_dimensions(i);

Completed in 1786 milliseconds