| /aosp_15_r20/external/tensorflow/tensorflow/lite/g3doc/guide/ |
| H A D | ops_version.md | 28 ### Change FlatBuffer schema
70 ### Change C structures and kernel implementation
108 ### Change the FlatBuffer reading code
153 ### Change kernel registration
182 ### Change TFLite op version
206 ### Update the operator version map
224 ### Delegation implementation
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/src/qnnpack/ |
| H A D | operator.h | 140 const struct pytorch_qnnp_operator* convolution) { in pytorch_qnnp_operator_get_log2_output_element_size()
145 const struct pytorch_qnnp_operator* convolution) { in pytorch_qnnp_operator_get_log2_input_element_size()
150 const struct pytorch_qnnp_operator* convolution) { in pytorch_qnnp_operator_get_log2_kernel_element_size()
155 const struct pytorch_qnnp_operator* convolution) { in pytorch_qnnp_operator_get_log2_bias_element_size()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | space_to_batch_converter.cc | 243 int64_t GetFirstChosenSpatialDim(HloInstruction* convolution) { in GetFirstChosenSpatialDim()
251 std::vector<int64_t> GetChosenSpatialDims(HloInstruction* convolution) { in GetChosenSpatialDims()
341 auto convolution = inst; in ConvolutionVisitor() local
373 HloInstruction* convolution, ConvolutionDimensionNumbers& dim_numbers) { in IsForwardWindowDilatedConv()
395 HloInstruction* convolution) { in IsConvSuitableForSpaceToBatch()
477 bool ConvolutionVisitor::IsThisBackPropFilterConv(HloInstruction* convolution) { in IsThisBackPropFilterConv()
2536 Status ConvolutionVisitor::PropagateOnConv(HloInstruction* convolution) { in PropagateOnConv()
2995 HloInstruction* convolution) { in PropagateOnBackpropFilterConv()
3543 HloInstruction* convolution, ConvolutionDimensionNumbers& dim_numbers) { in GetConvolutionDetails()
3626 HloInstruction* convolution) { in PerformSpaceToBatchOnConvolution()
|
| H A D | transpose_folding.cc | 42 const HloInstruction& convolution, in CanFoldOperandsIntoConvolution()
116 auto& convolution = *pair.first; in FoldTransposeIntoConvolution() local
|
| H A D | convolution_group_converter.cc | 204 Status ConvolutionVisitor::HandleBatchGroupCount(HloInstruction* convolution) { in HandleBatchGroupCount()
421 Status ConvolutionVisitor::HandleConvolution(HloInstruction* convolution) { in HandleConvolution()
|
| H A D | hlo_instruction.cc | 4391 if (auto* convolution = DynCast<HloConvolutionInstruction>(this)) { in precision_config() local
4405 if (auto* convolution = DynCast<HloConvolutionInstruction>(this)) { in mutable_precision_config() local
4719 if (auto convolution = DynCast<HloConvolutionInstruction>(this)) { in convolution_dimension_numbers() local
4730 if (auto convolution = DynCast<HloConvolutionInstruction>(this)) { in set_convolution_dimension_numbers() local
4740 if (auto convolution = DynCast<HloConvolutionInstruction>(this)) { in feature_group_count() local
4747 if (auto convolution = DynCast<HloConvolutionInstruction>(this)) { in set_feature_group_count() local
4755 if (auto convolution = DynCast<HloConvolutionInstruction>(this)) { in batch_group_count() local
4762 if (auto convolution = DynCast<HloConvolutionInstruction>(this)) { in set_batch_group_count() local
|
| H A D | hlo_cost_analysis.cc | 580 const HloInstruction* convolution) { in GetConvolutionFlops()
591 int64_t HloCostAnalysis::GetConvolutionFlops(const HloInstruction* convolution, in GetConvolutionFlops()
711 Status HloCostAnalysis::HandleConvolution(const HloInstruction* convolution) { in HandleConvolution()
|
| H A D | algebraic_simplifier.cc | 6279 HloInstruction* convolution) { in FoldConvInputPad()
6340 HloInstruction* convolution) { in FoldConvFilterPad()
6406 HloInstruction* convolution) { in SwapConvOperands()
6543 HloInstruction* convolution) { in SimplifyConvToDot()
6665 HloInstruction* convolution) { in HandleConvolution()
|
| H A D | dfs_hlo_visitor_with_default.h | 84 Status HandleConvolution(HloInstructionPtr convolution) override { in HandleConvolution()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/src/ |
| H A D | convolution.c | 71 pytorch_qnnp_operator_t convolution = NULL; in pytorch_qnnp_create_convolution_ndhwc_q8() local
660 pytorch_qnnp_operator_t convolution, in pytorch_qnnp_setup_convolution2d_nhwc_q8()
683 pytorch_qnnp_operator_t convolution, in pytorch_qnnp_setup_convolution_ndhwc_q8()
|
| H A D | conv-prepack.cc | 11 const pytorch_qnnp_operator_t convolution, in PrePackConvWeights()
|
| H A D | conv-run.cc | 333 const pytorch_qnnp_operator_t convolution, in qnnpackConv()
|
| /aosp_15_r20/external/mesa3d/src/gallium/drivers/etnaviv/ |
| H A D | etnaviv_ml_tp.c | 641 struct etna_operation *convolution, in etna_ml_lower_detranspose()
667 const struct pipe_ml_operation *convolution, in etna_ml_lower_reshuffle()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| H A D | gpu_hlo_cost_analysis.cc | 85 const HloInstruction* convolution) { in GetConvolutionFlops()
|
| H A D | ir_emitter.cc | 531 Status IrEmitter::HandleConvolution(HloInstruction* convolution) { in HandleConvolution()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| H A D | ir_emission_utils.cc | 43 const HloInstruction& convolution, in PotentiallyImplementedAsEigenConvolution()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/test/ |
| H A D | fully-connected-operator-tester.h | 216 pytorch_qnnp_operator_t convolution = nullptr; in testQ8() local
|
| H A D | convolution-operator-tester.h | 633 pytorch_qnnp_operator_t convolution = nullptr; variable
|
| /aosp_15_r20/external/mesa3d/src/gallium/targets/teflon/ |
| H A D | test_model_generation.py | 154 def convolution(batch_size, input_size, weight_size, in_ch, out_ch, stride, padding, signed, seed, … function
|
| /aosp_15_r20/external/pytorch/torch/_inductor/kernel/ |
| H A D | conv.py | 445 def convolution( function
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/mkldnn/ |
| H A D | ConvPrepack.cpp | 18 namespace convolution { namespace
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/passes/ |
| H A D | graph_rewrite_helper.cpp | 65 std::string convolution = R"( in replaceConvolutionWithAtenConv() local
|
| /aosp_15_r20/external/eigen/bench/tensors/ |
| H A D | tensor_benchmarks.h | 492 void convolution(int num_iters, int kernel_x, int kernel_y) { in convolution() function
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/mkldnn/xpu/detail/ |
| H A D | Conv.cpp | 158 sycl::event convolution( in convolution() function
|
| /aosp_15_r20/external/pytorch/torch/_inductor/fx_passes/ |
| H A D | mkldnn_fusion.py | 1033 def convolution(match, *args, **kwargs): function
|