| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_conv_rewriter.cc | 43 const ConvolutionDimensionNumbers& dnums, in CreateGpuConv()
122 const ConvolutionDimensionNumbers& dnums = in CanImplementAsGpuForwardConv()
146 std::tuple<bool, Window, ConvolutionDimensionNumbers, HloInstruction*>
150 std::make_tuple(false, Window(), ConvolutionDimensionNumbers(), nullptr); in MatchBackwardFilter()
179 const ConvolutionDimensionNumbers& conv_dnums = in MatchBackwardFilter()
307 ConvolutionDimensionNumbers backward_conv_dnums; in MatchBackwardFilter()
335 std::tuple<bool, Window, ConvolutionDimensionNumbers, HloInstruction*>
339 std::make_tuple(false, Window(), ConvolutionDimensionNumbers(), nullptr); in MatchBackwardInput()
344 ConvolutionDimensionNumbers dnums = conv->convolution_dimension_numbers(); in MatchBackwardInput()
619 ConvolutionDimensionNumbers dnums; in CreateCustomCallHelper()
|
| D | stream_executor_util.h | 43 StreamExecutorConvLayoutsToXlaLayouts(const ConvolutionDimensionNumbers& dnums,
51 XlaConvLayoutsToStreamExecutorLayouts(const ConvolutionDimensionNumbers& dnums,
|
| D | gpu_conv_padding_legalization.cc | 49 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()
|
| D | gpu_conv_runner.h | 133 ConvolutionDimensionNumbers dnums;
|
| D | gpu_conv_rewriter_test.cc | 94 ConvolutionDimensionNumbers tf_default_dnums_for_backward_filter_;
95 ConvolutionDimensionNumbers tf_default_dnums_for_backward_input_;
276 ConvolutionDimensionNumbers conv_dnums; in TEST_F()
|
| D | stream_executor_util.cc | 49 StreamExecutorConvLayoutsToXlaLayouts(const ConvolutionDimensionNumbers& dnums, in StreamExecutorConvLayoutsToXlaLayouts()
124 XlaConvLayoutsToStreamExecutorLayouts(const ConvolutionDimensionNumbers& dnums, in XlaConvLayoutsToStreamExecutorLayouts()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | convolution_dimension_numbers_test.cc | 37 StatusOr<ConvolutionDimensionNumbers> CreateConvDimensionNumbers( in CreateConvDimensionNumbers()
43 ConvolutionDimensionNumbers dimension_numbers; in CreateConvDimensionNumbers()
105 ConvolutionDimensionNumbers dim_nums = in XLA_TEST_F()
|
| D | convolution_test.cc | 254 ConvolutionDimensionNumbers dnums; in XLA_TEST_F()
320 ConvolutionDimensionNumbers dnums; in RunTest()
379 ConvolutionDimensionNumbers dnums; in RunTest()
445 ConvolutionDimensionNumbers dnums; in RunTest()
515 ConvolutionDimensionNumbers dnums; in RunTest()
579 ConvolutionDimensionNumbers dnums; in RunTest()
648 ConvolutionDimensionNumbers dnums; in RunTest()
717 ConvolutionDimensionNumbers dnums; in RunTest()
788 ConvolutionDimensionNumbers dnums; in RunTest()
863 ConvolutionDimensionNumbers dnums; in RunTest()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | convolution_4d_expander.cc | 39 const ConvolutionDimensionNumbers& dim_nums = in InstructionMatchesPattern()
59 ConvolutionDimensionNumbers dim_nums = in ExpandInstruction()
61 ConvolutionDimensionNumbers new_dim_nums = dim_nums; in ExpandInstruction()
|
| D | hlo_parser.h | 71 StatusOr<ConvolutionDimensionNumbers> ParseConvolutionDimensionNumbers(
|
| D | transpose_folding.cc | 156 const ConvolutionDimensionNumbers& dnums = in FoldTransposeIntoConvolution()
158 ConvolutionDimensionNumbers new_dnums = dnums; in FoldTransposeIntoConvolution()
|
| D | hlo_instructions.h | 1229 const ConvolutionDimensionNumbers& dimension_numbers,
1233 const ConvolutionDimensionNumbers& convolution_dimension_numbers() const { in convolution_dimension_numbers()
1237 const ConvolutionDimensionNumbers& dnums) { in set_convolution_dimension_numbers()
1285 ConvolutionDimensionNumbers convolution_dimension_numbers_;
1433 const ConvolutionDimensionNumbers& convolution_dimension_numbers() const { in convolution_dimension_numbers()
1439 const ConvolutionDimensionNumbers& dnums) { in set_convolution_dimension_numbers()
1441 absl::make_unique<ConvolutionDimensionNumbers>(dnums); in set_convolution_dimension_numbers()
1522 std::unique_ptr<ConvolutionDimensionNumbers> convolution_dimension_numbers_;
|
| D | space_to_batch_converter.cc | 80 ConvolutionDimensionNumbers& dim_numbers);
117 HloInstruction* activations, ConvolutionDimensionNumbers& dim_numbers,
162 HloInstruction* activations, ConvolutionDimensionNumbers& dim_numbers,
278 ConvolutionDimensionNumbers dim_numbers = in IsConvSuitableForSpaceToBatch()
479 HloInstruction* activations, ConvolutionDimensionNumbers& dim_numbers, in BringSpaceNextToBatch()
486 ConvolutionDimensionNumbers new_dim_numbers = dim_numbers; in BringSpaceNextToBatch()
2092 ConvolutionDimensionNumbers& dim_numbers, in SplitSpace()
2199 ConvolutionDimensionNumbers tmp_dim_numbers; in PropagateOnBackpropFilterConv()
2244 ConvolutionDimensionNumbers tmp_dim_numbers; in PropagateOnBackpropFilterConv()
2592 HloInstruction* convolution, ConvolutionDimensionNumbers& dim_numbers) { in GetConvolutionDetails()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | xla_client.py | 530 class ConvolutionDimensionNumbers(object): class
551 dimension_numbers: Union[None, ConvolutionDimensionNumbers, Tuple[str, str, argument
553 num_spatial_dimensions: int) -> ConvolutionDimensionNumbers:
582 dimension_numbers = ConvolutionDimensionNumbers()
594 dimension_numbers = ConvolutionDimensionNumbers()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | conv_canonicalization_test.cc | 70 ConvolutionDimensionNumbers dnums; in TEST_F()
133 ConvolutionDimensionNumbers dnums; in TEST_F()
|
| D | conv_canonicalization.cc | 39 const ConvolutionDimensionNumbers& dnums = in Run()
117 ConvolutionDimensionNumbers new_dnums; in Run()
|
| D | ir_emission_utils.cc | 83 const ConvolutionDimensionNumbers& dnums = in PotentiallyImplementedAsEigenConvolution()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.h | 253 static ConvolutionDimensionNumbers CreateDefaultConvDimensionNumbers(
257 static Status Validate(const ConvolutionDimensionNumbers& dnum);
549 Padding padding, const ConvolutionDimensionNumbers& dimension_numbers,
557 const ConvolutionDimensionNumbers& dimension_numbers,
567 const ConvolutionDimensionNumbers& dimension_numbers,
577 const ConvolutionDimensionNumbers& dimension_numbers,
588 const ConvolutionDimensionNumbers& dimension_numbers,
599 const ConvolutionDimensionNumbers& dimension_numbers,
609 const ConvolutionDimensionNumbers& dimension_numbers,
620 const ConvolutionDimensionNumbers& dimension_numbers,
[all …]
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | attribute_exporter.h | 29 ConvolutionDimensionNumbers ConvertConvDimensionNumbers(
|
| D | attribute_exporter.cc | 26 ConvolutionDimensionNumbers ConvertConvDimensionNumbers( in ConvertConvDimensionNumbers()
28 ConvolutionDimensionNumbers output; in ConvertConvDimensionNumbers()
|
| D | attribute_importer.h | 46 const xla::ConvolutionDimensionNumbers& dnums, mlir::Builder* builder);
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | xla_conv_op.cc | 85 xla::ConvolutionDimensionNumbers dnums_;
|
| D | conv_op_helpers.cc | 234 xla::ConvolutionDimensionNumbers dims; in MakeXlaForwardConvOp()
327 xla::ConvolutionDimensionNumbers dnums; in MakeXlaBackpropInputConvOp()
420 xla::ConvolutionDimensionNumbers dnums; in MakeXlaBackpropFilterConvOp()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | reference_util_test.cc | 333 ConvolutionDimensionNumbers dimension_numbers; in TEST_F()
389 ConvolutionDimensionNumbers dimension_numbers; in TEST_F()
|
| D | reference_util.cc | 59 const ConvolutionDimensionNumbers& dnums) { in ConvArray3DGeneralDimensionsDilated()
77 ConvolutionDimensionNumbers dnums2d = dnums; in ConvArray3DGeneralDimensionsDilated()
402 ConvolutionDimensionNumbers dimension_numbers) { in ConvArray4DGeneralDimensions()
413 ConvolutionDimensionNumbers dnums) { in ConvArray4DGeneralDimensionsDilated()
|