| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| ir_emission_utils.cc | 51 const ConvolutionDimensionNumbers& dnums = local
55 const int64 num_spatial_dims = dnums.output_spatial_dimensions_size();
61 if (dnums.input_spatial_dimensions(i) != i + 1) {
64 if (dnums.kernel_spatial_dimensions(i) != i) {
67 if (dnums.output_spatial_dimensions(i) != i + 1) {
73 return dnums.input_batch_dimension() == 0 &&
74 dnums.input_feature_dimension() == input_shape.dimensions_size() - 1 &&
75 dnums.output_batch_dimension() == 0 &&
76 dnums.output_feature_dimension() ==
78 dnums.kernel_input_feature_dimension() =
[all...] |
| conv_canonicalization.cc | 37 const ConvolutionDimensionNumbers& dnums = local
39 auto input_batch_dim = dnums.input_batch_dimension();
40 auto input_feature_dim = dnums.input_feature_dimension();
41 auto kernel_input_feature_dim = dnums.kernel_input_feature_dimension();
42 auto kernel_output_feature_dim = dnums.kernel_output_feature_dimension();
44 const int64 num_spatial_dims = dnums.output_spatial_dimensions_size();
63 new_input_dim_order[i + 1] = dnums.input_spatial_dimensions(i);
65 input->shape().dimensions(dnums.input_spatial_dimensions(i));
82 new_kernel_dim_order[i] = dnums.kernel_spatial_dimensions(i);
84 kernel->shape().dimensions(dnums.kernel_spatial_dimensions(i))
[all...] |
| conv_canonicalization_test.cc | 69 ConvolutionDimensionNumbers dnums; local
70 dnums.set_input_batch_dimension(1);
71 dnums.set_output_batch_dimension(1);
72 dnums.add_input_spatial_dimensions(2);
73 dnums.add_output_spatial_dimensions(2);
74 dnums.add_input_spatial_dimensions(3);
75 dnums.add_output_spatial_dimensions(3);
76 dnums.set_input_feature_dimension(0);
77 dnums.set_output_feature_dimension(0);
78 dnums.add_kernel_spatial_dimensions(2)
127 ConvolutionDimensionNumbers dnums; local
[all...] |
| ir_emitter.cc | 805 const DotDimensionNumbers& dnums = dot->dot_dimension_numbers(); local
854 const ConvolutionDimensionNumbers& dnums = local
874 const ConvolutionDimensionNumbers& dnums = local
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| transpose_folding.cc | 108 const ConvolutionDimensionNumbers& dnums = local
110 ConvolutionDimensionNumbers new_dnums = dnums;
124 transpose_dimensions[dnums.input_batch_dimension()]);
126 transpose_dimensions[dnums.input_feature_dimension()]);
148 transpose_dimensions[dnums.kernel_input_feature_dimension()]);
150 transpose_dimensions[dnums.kernel_output_feature_dimension()]);
|
| transpose_folding_test.cc | 225 auto dnums = ComputationBuilder::CreateDefaultConvDimensionNumbers(); local
235 transpose_y->shape().dimensions(dnums.kernel_spatial_dimensions(i)));
238 x->shape(), transpose_y->shape(), window, dnums);
241 conv_shape.ValueOrDie(), x, transpose_y, window, dnums));
258 EXPECT_EQ(dnums.kernel_input_feature_dimension(),
261 EXPECT_EQ(dnums.kernel_output_feature_dimension(),
278 auto dnums = ComputationBuilder::CreateDefaultConvDimensionNumbers(); local
288 transpose_y->shape().dimensions(dnums.kernel_spatial_dimensions(i)));
291 x->shape(), transpose_y->shape(), window, dnums);
294 conv_shape.ValueOrDie(), x, transpose_y, window, dnums));
337 auto dnums = ComputationBuilder::CreateDefaultConvDimensionNumbers(); local
401 auto dnums = ComputationBuilder::CreateDefaultConvDimensionNumbers(); local
[all...] |
| dot_decomposer.cc | 34 const DotDimensionNumbers& dnums = dot->dot_dimension_numbers(); local
42 CHECK_EQ(dnums.lhs_batch_dimensions_size(),
43 dnums.rhs_batch_dimensions_size());
44 const int64 num_batch_dims = dnums.lhs_batch_dimensions_size();
49 CHECK_EQ(lhs_shape.dimensions(dnums.lhs_batch_dimensions(i)),
50 rhs_shape.dimensions(dnums.rhs_batch_dimensions(i)));
51 batch_size *= lhs_shape.dimensions(dnums.lhs_batch_dimensions(i));
55 CHECK_EQ(1, dnums.lhs_contracting_dimensions_size());
56 const int64 lhs_contracting_dim_number = dnums.lhs_contracting_dimensions(0);
59 CHECK_EQ(1, dnums.rhs_contracting_dimensions_size())
169 const DotDimensionNumbers& dnums = instruction->dot_dimension_numbers(); local
[all...] |
| hlo_cost_analysis.cc | 204 const DotDimensionNumbers& dnums = dot->dot_dimension_numbers(); local
208 lhs_shape.dimensions(dnums.lhs_contracting_dimensions(0));
383 const auto& dnums = convolution->convolution_dimension_numbers(); local
385 convolution->shape().dimensions(dnums.output_feature_dimension());
|
| hlo_evaluator_test.cc | 765 ConvolutionDimensionNumbers dnums; local
766 dnums.set_input_batch_dimension(0);
767 dnums.set_output_batch_dimension(0);
768 dnums.set_input_feature_dimension(1);
769 dnums.set_output_feature_dimension(1);
770 dnums.add_input_spatial_dimensions(2);
771 dnums.add_output_spatial_dimensions(2);
773 dnums.set_kernel_output_feature_dimension(0);
774 dnums.set_kernel_input_feature_dimension(1);
775 dnums.add_kernel_spatial_dimensions(2)
828 ConvolutionDimensionNumbers dnums = local
899 ConvolutionDimensionNumbers dnums; local
976 ConvolutionDimensionNumbers dnums; local
1047 ConvolutionDimensionNumbers dnums = local
1110 ConvolutionDimensionNumbers dnums = local
1181 ConvolutionDimensionNumbers dnums = local
[all...] |
| shape_inference_test.cc | 391 ConvolutionDimensionNumbers dnums; local
395 dnums.set_input_batch_dimension(0);
396 dnums.set_output_batch_dimension(0);
397 dnums.set_input_feature_dimension(1);
398 dnums.set_output_feature_dimension(1);
399 dnums.add_input_spatial_dimensions(2);
400 dnums.add_output_spatial_dimensions(2);
401 dnums.add_input_spatial_dimensions(3);
402 dnums.add_output_spatial_dimensions(3);
406 dnums.set_kernel_input_feature_dimension(2)
435 ConvolutionDimensionNumbers dnums; local
480 ConvolutionDimensionNumbers dnums; local
529 ConvolutionDimensionNumbers dnums; local
[all...] |
| algebraic_simplifier.cc | 782 const DotDimensionNumbers& dnums = dot->dot_dimension_numbers(); local
783 if (dnums.lhs_contracting_dimensions_size() != 1 ||
784 dnums.lhs_batch_dimensions_size() != 0) {
788 const int64 lhs_contracting_dim = dnums.lhs_contracting_dimensions(0);
789 const int64 rhs_contracting_dim = dnums.rhs_contracting_dimensions(0);
1884 const ConvolutionDimensionNumbers& dnums = local
[all...] |
| algebraic_simplifier_test.cc | 889 ConvolutionDimensionNumbers dnums; local
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| cudnn_convolution_rewriter.cc | 38 const ConvolutionDimensionNumbers& dnums = local
40 if (dnums.input_spatial_dimensions_size() > 3) {
226 ConvolutionDimensionNumbers dnums = conv->convolution_dimension_numbers(); local
227 const auto& kernel_spatial_dims = dnums.kernel_spatial_dimensions();
274 const auto& input_spatial_dims = dnums.input_spatial_dimensions();
275 const auto& output_spatial_dims = dnums.output_spatial_dimensions();
387 dnums.set_kernel_input_feature_dimension(
389 dnums.set_kernel_output_feature_dimension(
392 return std::make_tuple(true, new_window, dnums);
402 ConvolutionDimensionNumbers dnums;
[all...] |
| ir_emitter_unnested.cc | 294 const DotDimensionNumbers& dnums = dot->dot_dimension_numbers(); local
295 if (dnums.lhs_batch_dimensions_size() > 0 ||
296 dnums.rhs_batch_dimensions_size() > 0) {
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| convolution_variants_test.cc | 981 ConvolutionDimensionNumbers dnums; local
1025 ConvolutionDimensionNumbers dnums; local
1069 ConvolutionDimensionNumbers dnums; local
1110 ConvolutionDimensionNumbers dnums; local
[all...] |
| convolution_test.cc | 430 ConvolutionDimensionNumbers dnums; local
431 dnums.set_input_batch_dimension(0);
432 dnums.set_output_batch_dimension(0);
433 dnums.add_input_spatial_dimensions(1);
434 dnums.add_output_spatial_dimensions(1);
435 dnums.add_input_spatial_dimensions(2);
436 dnums.add_output_spatial_dimensions(2);
437 dnums.add_input_spatial_dimensions(3);
438 dnums.add_output_spatial_dimensions(3);
439 dnums.set_input_feature_dimension(4)
497 ConvolutionDimensionNumbers dnums; local
562 ConvolutionDimensionNumbers dnums; local
622 ConvolutionDimensionNumbers dnums; local
[all...] |
| dot_operation_test.cc | 534 DotDimensionNumbers dnums; local
535 dnums.add_lhs_contracting_dimensions(2);
536 dnums.add_rhs_contracting_dimensions(1);
537 dnums.add_lhs_batch_dimensions(0);
538 dnums.add_rhs_batch_dimensions(0);
540 auto out = builder.DotGeneral(x, y, dnums);
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| conv_ops.cc | 398 xla::ConvolutionDimensionNumbers dnums; variable
399 dnums.set_input_batch_dimension(batch_dim);
400 dnums.set_output_batch_dimension(batch_dim);
401 dnums.set_input_feature_dimension(feature_dim);
402 dnums.set_output_feature_dimension(feature_dim);
406 dnums.set_kernel_input_feature_dimension(num_spatial_dims_ + 1);
407 dnums.set_kernel_output_feature_dimension(num_spatial_dims_);
416 dnums.add_input_spatial_dimensions(dim);
417 dnums.add_kernel_spatial_dimensions(i);
418 dnums.add_output_spatial_dimensions(dim)
553 xla::ConvolutionDimensionNumbers dnums; variable
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/tools/parser/ |
| hlo_parser.cc | 167 bool ParseConvolutionDimensionNumbers(ConvolutionDimensionNumbers* dnums);
686 optional<ConvolutionDimensionNumbers> dnums; local
689 AttrTy::kConvolutionDimensionNumbers, &dnums};
698 shape, /*lhs=*/operands[0], /*rhs=*/operands[1], *window, *dnums));
[all...] |
