Home
last modified time | relevance | path

Searched defs:conv_op (Results 1 – 25 of 31) sorted by relevance

12

/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/expansions/
H A Dconv_spmd_expander.cc42 ConvOp conv_op) { in VerifyConvLayout()
139 StatusOr<mlir::Operation*> HandleConv(ConvOp conv_op) { in HandleConv()
290 const Layout& output_layout, ConvBackpropInputOp conv_op) { in HandleConvBackpropInput()
310 const Layout& output_layout, ConvBackpropFilterOp conv_op) { in HandleConvBackpropFilter()
/aosp_15_r20/external/tensorflow/tensorflow/tools/graph_transforms/
H A Dfold_old_batch_norms_test.cc57 Output conv_op = Conv2D(root.WithOpName("conv_op"), input_op, weights_op, in TestFoldOldBatchNorms() local
141 Output conv_op = DepthwiseConv2dNative(root.WithOpName("conv_op"), input_op, in TestFoldOldBatchNormsAfterDepthwiseConv2dNative() local
219 Output conv_op = Conv2D(root.WithOpName("conv_op"), input_op, weights_op, in TestFoldFusedBatchNorms() local
296 Output conv_op = DepthwiseConv2dNative(root.WithOpName("conv_op"), input_op, in TestFoldFusedBatchNormsAfterDepthwiseConv2dNative() local
475 Output conv_op = Conv2D(root.WithOpName("conv_op"), input_op, weights_op, in TestFoldFusedBatchNormsWithBatchToSpace() local
H A Dfuse_convolutions_test.cc72 Output conv_op = Conv2D(root.WithOpName("output"), pad_op, weights_op, in TestFuseResizePadAndConv() local
122 Output conv_op = Conv2D(root.WithOpName("output"), resize_op, weights_op, in TestFuseResizeAndConv() local
174 Output conv_op = Conv2D(root.WithOpName("output"), pad_op, weights_op, in TestFusePadAndConv() local
H A Dfold_batch_norms_test.cc56 Output conv_op = Conv2D(root.WithOpName("conv_op"), input_op, weights_op, in TestFoldBatchNormsConv2D() local
107 Output conv_op = DepthwiseConv2dNative(root.WithOpName("conv_op"), input_op, in TestFoldBatchNormsDepthwiseConv2dNative() local
158 Output conv_op = Conv2D(root.WithOpName("conv_op"), input_op, weights_op, in TestFoldBatchNormsConv2DShared() local
H A Dround_weights_test.cc59 Output conv_op = Conv2D(root.WithOpName("output"), input_op, weights_op, in TestRoundWeights() local
H A Dflatten_atrous_test.cc70 Output conv_op = TConvOp(root.WithOpName("conv_op"), space_to_batch_op, in TestFlattenAtrousConv() local
H A Dquantize_weights_test.cc55 Output conv_op = ops::Conv2D(root.WithOpName("output"), input_op, in BuildGraphDef() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/core/api/
H A Dflatbuffer_conversions_test.cc102 const Operator* conv_op = in TEST_F() local
122 const Operator* conv_op = BuildTestOperator( in TEST_F() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/
H A Dquantization_utils_test.cc544 auto conv_op = subgraph->operators.at(0).get(); in TEST_F() local
594 auto conv_op = subgraph->operators.at(0).get(); in TEST_F() local
629 auto conv_op = subgraph->operators.at(0).get(); in TEST_F() local
697 auto conv_op = subgraph->operators.at(0).get(); in TEST_F() local
H A Dquantize_model_test.cc624 auto conv_op = subgraph->operators[0].get(); in TEST_F() local
735 auto conv_op = subgraph->operators[0].get(); in TEST_P() local
839 auto conv_op = subgraph->operators[0].get(); in TEST_P() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
H A Dlegalize_hlo.cc201 mhlo::ConvolutionOp conv_op, OpAdaptor adaptor, in matchAndRewrite()
398 mhlo::ConvolutionOp conv_op, OpAdaptor adaptor, in matchAndRewrite()
472 bool IsSamePadding(mhlo::ConvolutionOp conv_op, int num_spatial_dims, in IsSamePadding()
605 void CreateConvOp(mhlo::ConvolutionOp conv_op, ArrayRef<int64_t> strides, in CreateConvOp()
707 mhlo::ConvolutionOp conv_op, OpAdaptor adaptor, in matchAndRewrite()
792 bool IsSamePadding(mhlo::ConvolutionOp conv_op, int num_spatial_dims, in IsSamePadding()
808 LogicalResult IsSupportedConvOp(mhlo::ConvolutionOp conv_op, in IsSupportedConvOp()
892 void CreateResizeBilinearOp(mhlo::ConvolutionOp conv_op, in CreateResizeBilinearOp()
912 LogicalResult MatchResizeOp(mhlo::ConvolutionOp conv_op, bool &align_corners, in MatchResizeOp()
/aosp_15_r20/external/tensorflow/tensorflow/lite/toco/graph_transformations/
H A Ddrop_im2col_arrays.cc29 auto* conv_op = static_cast<ConvOperator*>(conv_it->get()); in Run() local
H A Dconvert_pure_conv_to_depthwise.cc35 const auto* conv_op = static_cast<ConvOperator*>(conv_it->get()); in Run() local
H A Didentify_dilated_conv.cc63 auto* conv_op = static_cast<T*>(conv_base_op); in ResolveDilatedConv() local
/aosp_15_r20/external/executorch/backends/qualcomm/builders/
H A Dop_conv2d.py35 def _add_conv_op_parameter( argument
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cudnn/
H A DConv.cpp148 …auto conv_op = cudnn_frontend::OperationBuilder(CUDNN_BACKEND_OPERATION_CONVOLUTION_FORWARD_DESCRI… in apply_impl_helper() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/quantization/lite/
H A Dquantize_model_test.cc428 auto conv_op = subgraph->operators[0].get(); in TEST_P() local
533 auto conv_op = subgraph->operators[0].get(); in TEST_P() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/toco/
H A Ddump_graphviz.cc378 const auto& conv_op = static_cast<const ConvOperator&>(op); in GetOpAttributes() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/
H A Dconv_2d_builder.cc179 auto* conv_op = graph_builder_->AddNode(GetTFLiteNodeID()); in BuildDilatedDwConv() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/
H A Doperation_selector.cc168 auto& conv_op = gpu_subgraph->operations.back(); in AddDynamicConv() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/experimental/tac/transforms/
H A Ddevice_transform_patterns.cc360 TFL::Conv2DOp conv_op, PatternRewriter& rewriter) const { in matchAndRewrite()
/aosp_15_r20/external/tensorflow/tensorflow/lite/toco/tflite/
H A Dexport_test.cc534 const TocoOperator& conv_op = in GetVersion() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
H A Dprepare_tf.cc267 auto conv_op = static_cast<const ConcreteType *>(this)->createTFLOp( in matchAndRewrite() local
/aosp_15_r20/external/pytorch/test/quantization/core/
H A Dtest_quantized_op.py5009 self, qconv_fn, qconv_prepack_fn, conv_op, batch_size, argument
6634 conv_op, argument
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cudnn/
H A DConv_v8.cpp492 auto conv_op = in build_opgraph_fused() local

12