Home
last modified time | relevance | path

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

/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/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/delegates/gpu/common/selectors/
Doperation_selector.cc222 auto& conv_op = gpu_subgraph->operations[2]; in GPUOperationFromNode() local
299 auto& conv_op = gpu_subgraph->operations[1]; in GPUOperationFromNode() local
347 auto& conv_op = gpu_subgraph->operations[1]; in GPUOperationFromNode() local
/external/tensorflow/tensorflow/lite/tools/optimize/
Dquantization_utils_test.cc474 auto conv_op = subgraph->operators.at(0).get(); in TEST_F() local
524 auto conv_op = subgraph->operators.at(0).get(); in TEST_F() local
559 auto conv_op = subgraph->operators.at(0).get(); in TEST_F() local
628 auto conv_op = subgraph->operators.at(0).get(); in TEST_F() local
Dquantize_model_test.cc609 auto conv_op = subgraph->operators[0].get(); in TEST_F() local
718 auto conv_op = subgraph->operators[0].get(); in TEST_P() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dlegalize_hlo.cc65 mhlo::ConvOp conv_op, ArrayRef<Value> args, in matchAndRewrite()
136 bool IsSamePadding(mhlo::ConvOp conv_op, int num_spatial_dims, in IsSamePadding()
158 void CreateConvOp(mhlo::ConvOp conv_op, ArrayRef<int64_t> strides, in CreateConvOp()
/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/
Dconv_2d_builder.cc270 auto* conv_op = graph_builder_->AddNode(GetTFLiteNodeID()); in PopulateSubGraph() local
/external/tensorflow/tensorflow/lite/toco/
Ddump_graphviz.cc376 const auto& conv_op = static_cast<const ConvOperator&>(op); in GetOpAttributes() local
/external/tensorflow/tensorflow/lite/toco/tflite/
Dexport_test.cc533 const TocoOperator& conv_op = in GetVersion() local
Doperator_test.cc1210 ConvOperator conv_op; in TEST_F() local
Doperator.cc180 const auto& conv_op = in GetVersion() local
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dprepare_tf.cc369 auto conv_op = static_cast<const ConcreteType *>(this)->createTFLOp( in matchAndRewrite() local
/external/tensorflow/tensorflow/python/keras/layers/
Dconvolutional.py133 conv_op=None, argument
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc1071 miopenFusionOpDescriptor_t conv_op; in SetConvolutionArgs() local
1245 miopenFusionOpDescriptor_t conv_op; in ScopedFusionPlanConvolutionBiasActivation() local