Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/tools/graph_transforms/
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
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
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
Dround_weights_test.cc59 Output conv_op = Conv2D(root.WithOpName("output"), input_op, weights_op, in TestRoundWeights() local
Dflatten_atrous_test.cc70 Output conv_op = TConvOp(root.WithOpName("conv_op"), space_to_batch_op, in TestFlattenAtrousConv() local
Dquantize_weights_test.cc55 Output conv_op = ops::Conv2D(root.WithOpName("output"), input_op, in BuildGraphDef() local
Dquantize_nodes_test.cc335 Output conv_op = Conv2D(root.WithOpName("conv_op"), input_op, filter_op, in TestQuantizeConv2D() local
/external/tensorflow/tensorflow/dtensor/mlir/expansions/
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()
/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Ddrop_im2col_arrays.cc29 auto* conv_op = static_cast<ConvOperator*>(conv_it->get()); in Run() local
Dconvert_pure_conv_to_depthwise.cc35 const auto* conv_op = static_cast<ConvOperator*>(conv_it->get()); in Run() local
Didentify_dilated_conv.cc63 auto* conv_op = static_cast<T*>(conv_base_op); in ResolveDilatedConv() local
/external/tensorflow/tensorflow/lite/core/api/
Dflatbuffer_conversions_test.cc102 const Operator* conv_op = in TEST_F() local
122 const Operator* conv_op = BuildTestOperator( in TEST_F() local
/external/tensorflow/tensorflow/lite/tools/optimize/
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
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
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
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()
/external/tensorflow/tensorflow/lite/toco/
Ddump_graphviz.cc378 const auto& conv_op = static_cast<const ConvOperator&>(op); in GetOpAttributes() local
/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/
Dconv_2d_builder.cc179 auto* conv_op = graph_builder_->AddNode(GetTFLiteNodeID()); in BuildDilatedDwConv() local
/external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/
Doperation_selector.cc168 auto& conv_op = gpu_subgraph->operations.back(); in AddDynamicConv() local
/external/tensorflow/tensorflow/compiler/mlir/lite/quantization/lite/
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
/external/tensorflow/tensorflow/compiler/mlir/lite/experimental/tac/transforms/
Ddevice_transform_patterns.cc360 TFL::Conv2DOp conv_op, PatternRewriter& rewriter) const { in matchAndRewrite()
/external/tensorflow/tensorflow/lite/toco/tflite/
Dexport_test.cc534 const TocoOperator& conv_op = in GetVersion() local
Doperator_test.cc1245 ConvOperator conv_op; in TEST_F() local
Doperator.cc191 const auto& conv_op = in GetVersion() local
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dprepare_tf.cc267 auto conv_op = static_cast<const ConcreteType *>(this)->createTFLOp( in matchAndRewrite() local
/external/tensorflow/tensorflow/python/keras/layers/
Dconvolutional.py128 conv_op=None, argument

12