Home
last modified time | relevance | path

Searched refs:ConvolutionDimensionNumbers (Results 1 – 25 of 47) sorted by relevance

12

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_conv_rewriter.cc42 const ConvolutionDimensionNumbers& dnums, in CreateCudnnConv()
68 const ConvolutionDimensionNumbers& dnums = in CanImplementAsCudnnForwardConv()
92 std::tuple<bool, Window, ConvolutionDimensionNumbers> MatchBackwardFilter( in MatchBackwardFilter()
95 std::make_tuple(false, Window(), ConvolutionDimensionNumbers()); in MatchBackwardFilter()
116 const ConvolutionDimensionNumbers& conv_dnums = in MatchBackwardFilter()
229 ConvolutionDimensionNumbers backward_conv_dnums; in MatchBackwardFilter()
256 std::tuple<bool, Window, ConvolutionDimensionNumbers, HloInstruction*>
259 std::make_tuple(false, Window(), ConvolutionDimensionNumbers(), nullptr); in MatchBackwardInput()
273 ConvolutionDimensionNumbers dnums = conv->convolution_dimension_numbers(); in MatchBackwardInput()
504 ConvolutionDimensionNumbers dnums; in RunOnInstruction()
Dstream_executor_util.h36 StreamExecutorConvLayoutsToXlaLayouts(const ConvolutionDimensionNumbers& dnums,
44 XlaConvLayoutsToStreamExecutorLayouts(const ConvolutionDimensionNumbers& dnums,
Dcudnn_conv_padding_legalization.cc49 Window* conv_window, const ConvolutionDimensionNumbers& conv_dnums, in MaybePaddedAndSlicedInput()
124 const ConvolutionDimensionNumbers& conv_dnums, in MaybePaddedKernel()
223 ConvolutionDimensionNumbers backward_conv_dnums = in CanonicalizeBackwardFilterConvolution()
282 ConvolutionDimensionNumbers backward_conv_dnums = in CanonicalizeBackwardInputConvolution()
Dstream_executor_util.cc38 StreamExecutorConvLayoutsToXlaLayouts(const ConvolutionDimensionNumbers& dnums, in StreamExecutorConvLayoutsToXlaLayouts()
113 XlaConvLayoutsToStreamExecutorLayouts(const ConvolutionDimensionNumbers& dnums, in XlaConvLayoutsToStreamExecutorLayouts()
Dcudnn_conv_rewriter_test.cc94 ConvolutionDimensionNumbers tf_default_dnums_for_backward_filter_;
95 ConvolutionDimensionNumbers tf_default_dnums_for_backward_input_;
274 ConvolutionDimensionNumbers conv_dnums; in TEST_F()
Dcudnn_conv_runner.cc70 const ConvolutionDimensionNumbers* dnums;
125 const ConvolutionDimensionNumbers& dnums = *params.dnums; in RunCudnnConvImpl()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dconvolution_dimension_numbers_test.cc37 StatusOr<ConvolutionDimensionNumbers> CreateConvDimensionNumbers( in CreateConvDimensionNumbers()
43 ConvolutionDimensionNumbers dimension_numbers; in CreateConvDimensionNumbers()
105 ConvolutionDimensionNumbers dim_nums = in XLA_TEST_F()
Dconvolution_test.cc422 ConvolutionDimensionNumbers dnums; in XLA_TEST_F()
488 ConvolutionDimensionNumbers dnums; in RunTest()
547 ConvolutionDimensionNumbers dnums; in RunTest()
613 ConvolutionDimensionNumbers dnums; in RunTest()
683 ConvolutionDimensionNumbers dnums; in RunTest()
747 ConvolutionDimensionNumbers dnums; in RunTest()
816 ConvolutionDimensionNumbers dnums; in RunTest()
885 ConvolutionDimensionNumbers dnums; in RunTest()
956 ConvolutionDimensionNumbers dnums; in RunTest()
1031 ConvolutionDimensionNumbers dnums; in RunTest()
[all …]
Dconvolution_variants_test.cc980 ConvolutionDimensionNumbers dnums; in XLA_TEST_F()
1024 ConvolutionDimensionNumbers dnums; in XLA_TEST_F()
1068 ConvolutionDimensionNumbers dnums; in XLA_TEST_F()
1109 ConvolutionDimensionNumbers dnums; in XLA_TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dconv_canonicalization.cc38 const ConvolutionDimensionNumbers& dnums = in Run()
116 ConvolutionDimensionNumbers new_dnums; in Run()
Dconv_canonicalization_test.cc70 ConvolutionDimensionNumbers dnums; in TEST_F()
133 ConvolutionDimensionNumbers dnums; in TEST_F()
Dir_emission_utils.cc83 const ConvolutionDimensionNumbers& dnums = in PotentiallyImplementedAsEigenConvolution()
/external/tensorflow/tensorflow/compiler/xla/service/
Dtranspose_folding.cc127 const ConvolutionDimensionNumbers& dnums = in FoldTransposeIntoConvolution()
129 ConvolutionDimensionNumbers new_dnums = dnums; in FoldTransposeIntoConvolution()
Dhlo_parser.h62 StatusOr<ConvolutionDimensionNumbers> ParseConvolutionDimensionNumbers(
Dhlo_instructions.h1034 const ConvolutionDimensionNumbers& dimension_numbers,
1038 const ConvolutionDimensionNumbers& convolution_dimension_numbers() const { in convolution_dimension_numbers()
1042 const ConvolutionDimensionNumbers& dnums) { in set_convolution_dimension_numbers()
1087 ConvolutionDimensionNumbers convolution_dimension_numbers_;
1191 const ConvolutionDimensionNumbers& convolution_dimension_numbers() const { in convolution_dimension_numbers()
1197 const ConvolutionDimensionNumbers& dnums) { in set_convolution_dimension_numbers()
1199 absl::make_unique<ConvolutionDimensionNumbers>(dnums); in set_convolution_dimension_numbers()
1242 std::unique_ptr<ConvolutionDimensionNumbers> convolution_dimension_numbers_;
Dhlo_creation_utils.h59 const Window& window, const ConvolutionDimensionNumbers& dimension_numbers,
Dhlo_instruction.h440 const ConvolutionDimensionNumbers& dimension_numbers,
1549 const ConvolutionDimensionNumbers& convolution_dimension_numbers() const;
1555 const ConvolutionDimensionNumbers& dnums);
1816 const ConvolutionDimensionNumbers& dnums);
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.h189 static ConvolutionDimensionNumbers CreateDefaultConvDimensionNumbers(
193 static Status Validate(const ConvolutionDimensionNumbers& dnum);
398 const ConvolutionDimensionNumbers& dimension_numbers,
405 const ConvolutionDimensionNumbers& dimension_numbers,
414 const ConvolutionDimensionNumbers& dimension_numbers,
644 const ConvolutionDimensionNumbers& dimension_numbers) const;
794 const ConvolutionDimensionNumbers& dimension_numbers,
800 const ConvolutionDimensionNumbers& dimension_numbers,
809 const ConvolutionDimensionNumbers& dimension_numbers,
1322 Padding padding, const ConvolutionDimensionNumbers& dimension_numbers,
[all …]
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dxla_conv_op.cc85 xla::ConvolutionDimensionNumbers dnums_;
Dconv_op_helpers.cc285 xla::ConvolutionDimensionNumbers dims; in MakeXlaForwardConvOp()
351 xla::ConvolutionDimensionNumbers dnums; in MakeXlaBackpropInputConvOp()
423 xla::ConvolutionDimensionNumbers dnums; in MakeXlaBackpropFilterConvOp()
Dextract_image_patches_op.cc120 xla::ConvolutionDimensionNumbers dims; in Compile()
/external/tensorflow/tensorflow/compiler/xla/
Dreference_util_test.cc333 ConvolutionDimensionNumbers dimension_numbers; in TEST_F()
389 ConvolutionDimensionNumbers dimension_numbers; in TEST_F()
Dreference_util.cc74 const ConvolutionDimensionNumbers& dnums) { in ConvArray3DGeneralDimensionsDilated()
92 ConvolutionDimensionNumbers dnums2d = dnums; in ConvArray3DGeneralDimensionsDilated()
468 ConvolutionDimensionNumbers dimension_numbers) { in ConvArray4DGeneralDimensions()
479 ConvolutionDimensionNumbers dnums) { in ConvArray4DGeneralDimensionsDilated()
Dreference_util.h80 ConvolutionDimensionNumbers dimension_numbers);
88 std::pair<int64, int64> rhs_dilation, ConvolutionDimensionNumbers dnums);
102 const ConvolutionDimensionNumbers& dnums);
/external/tensorflow/tensorflow/compiler/xla/python/
Dxla_data.i521 %typemap(in) const ConvolutionDimensionNumbers&
522 (ConvolutionDimensionNumbers dimension_numbers) {

12