HomeSort by relevance Sort by last modified time
    Searched defs:dimension_numbers (Results 1 - 4 of 4) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/
reference_util_test.cc 333 ConvolutionDimensionNumbers dimension_numbers; local
334 dimension_numbers.set_input_batch_dimension(2);
335 dimension_numbers.set_input_feature_dimension(0);
336 dimension_numbers.set_output_batch_dimension(2);
337 dimension_numbers.set_output_feature_dimension(0);
338 dimension_numbers.add_input_spatial_dimensions(1);
339 dimension_numbers.add_output_spatial_dimensions(1);
340 dimension_numbers.add_input_spatial_dimensions(3);
341 dimension_numbers.add_output_spatial_dimensions(3);
342 dimension_numbers.set_kernel_output_feature_dimension(0)
389 ConvolutionDimensionNumbers dimension_numbers; local
    [all...]
  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
gpu_layout_assignment.cc 74 const ConvolutionDimensionNumbers& dimension_numbers = local
77 for (int i = dimension_numbers.input_spatial_dimensions_size() - 1; i >= 0;
79 input_layout.push_back(dimension_numbers.input_spatial_dimensions(i));
81 input_layout.push_back(dimension_numbers.input_feature_dimension());
82 input_layout.push_back(dimension_numbers.input_batch_dimension());
86 for (int i = dimension_numbers.kernel_spatial_dimensions_size() - 1; i >= 0;
88 filter_layout.push_back(dimension_numbers.kernel_spatial_dimensions(i));
90 filter_layout.push_back(dimension_numbers.kernel_input_feature_dimension());
91 filter_layout.push_back(dimension_numbers.kernel_output_feature_dimension());
95 for (int i = dimension_numbers.output_spatial_dimensions_size() - 1; i >= 0
    [all...]
  /external/tensorflow/tensorflow/compiler/tf2xla/kernels/
image_resize_ops.cc 156 xla::ConvolutionDimensionNumbers dimension_numbers; local
157 dimension_numbers.set_input_batch_dimension(0);
158 dimension_numbers.set_output_batch_dimension(0);
159 dimension_numbers.set_input_feature_dimension(3);
160 dimension_numbers.set_output_feature_dimension(3);
162 dimension_numbers.add_input_spatial_dimensions(1 + i);
163 dimension_numbers.add_output_spatial_dimensions(1 + i);
164 dimension_numbers.add_kernel_spatial_dimensions(i);
166 dimension_numbers.set_kernel_input_feature_dimension(num_spatial_dims);
167 dimension_numbers.set_kernel_output_feature_dimension(num_spatial_dims + 1)
201 xla::ConvolutionDimensionNumbers dimension_numbers; local
    [all...]
  /external/tensorflow/tensorflow/compiler/xla/client/
computation_builder.cc 540 DotDimensionNumbers dimension_numbers; local
541 dimension_numbers.add_lhs_contracting_dimensions(
543 dimension_numbers.add_rhs_contracting_dimensions(0);
544 return DotGeneral(lhs, rhs, dimension_numbers);
549 const DotDimensionNumbers& dimension_numbers) {
554 *request->mutable_dimension_numbers() = dimension_numbers;
576 const ConvolutionDimensionNumbers& dimension_numbers) {
618 dimension_numbers.input_spatial_dimensions()) &&
621 dimension_numbers.kernel_spatial_dimensions()) &&
624 dimension_numbers.output_spatial_dimensions())
1514 ConvolutionDimensionNumbers dimension_numbers; local
1564 ConvolutionDimensionNumbers dimension_numbers; local
    [all...]

Completed in 270 milliseconds