Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dconvert_pure_conv_to_depthwise.cc35 const auto* conv_op = static_cast<ConvOperator*>(conv_it->get()); in Run() local
36 if (conv_op->stride_width != conv_op->stride_height) { in Run()
39 if ((conv_op->dilation_width_factor != 1) || in Run()
40 (conv_op->dilation_height_factor != 1)) { in Run()
44 auto& input_array = model->GetArray(conv_op->inputs[0]); in Run()
55 const auto& weights_name = conv_op->inputs[1]; in Run()
61 LogName(*conv_op)); in Run()
76 LogName(*conv_op)); in Run()
79 depthwiseconv_op->inputs = conv_op->inputs; in Run()
81 depthwiseconv_op->outputs = {conv_op->outputs[0]}; in Run()
[all …]
Didentify_dilated_conv.cc60 auto* conv_op = static_cast<T*>(conv_base_op); in ResolveDilatedConv() local
61 if (conv_op->inputs.size() != 2) { in ResolveDilatedConv()
65 CHECK_EQ(conv_op->outputs.size(), 1); in ResolveDilatedConv()
68 auto* post_conv_op = GetOpWithInput(*model, conv_op->outputs[0]); in ResolveDilatedConv()
84 : GetOpWithInput(*model, conv_op->outputs[0]); in ResolveDilatedConv()
98 : GetOpWithInput(*model, conv_op->outputs[0]); in ResolveDilatedConv()
125 conv_op->dilation_width_factor = dilation_factor; in ResolveDilatedConv()
126 conv_op->dilation_height_factor = dilation_factor; in ResolveDilatedConv()
127 conv_op->padding.type = PaddingType::kSame; in ResolveDilatedConv()
133 post_conv_op->inputs[0] = conv_op->outputs[0]; in ResolveDilatedConv()
[all …]
Ddrop_im2col_arrays.cc29 auto* conv_op = static_cast<ConvOperator*>(conv_it->get()); in Run() local
30 if (conv_op->outputs.size() < 2) { in Run()
36 CHECK_EQ(conv_op->outputs.size(), 2); in Run()
37 model->EraseArray(conv_op->outputs[1]); in Run()
38 conv_op->outputs.resize(1); in Run()
39 AddMessageF("Dropped an im2col array for %s", LogName(*conv_op)); in Run()
/external/tensorflow/tensorflow/python/tools/
Doptimize_for_inference_lib.py240 conv_op = node_from_map(input_node_map,
242 if conv_op.op != "Conv2D":
247 weights_op = node_from_map(input_node_map, conv_op.input[1])
251 " run first?" % (conv_op.name, weights_op))
319 nodes_to_skip[conv_op.name] = True
344 new_conv_op.CopyFrom(conv_op)
347 offset_op.name = conv_op.name + "_bn_offset"
355 bias_add_op.attr["T"].CopyFrom(conv_op.attr["T"])
356 bias_add_op.attr["data_format"].CopyFrom(conv_op.attr["data_format"])
412 conv_op = node
[all …]
Doptimize_for_inference_test.py140 conv_op = nn_ops.conv2d(
152 conv_op,
187 conv_op = nn_ops.conv2d(
203 conv_op,
/external/tensorflow/tensorflow/tools/graph_transforms/
Dfold_batch_norms_test.cc56 Output conv_op = Conv2D(root.WithOpName("conv_op"), input_op, weights_op, in TestFoldBatchNormsConv2D() local
64 Output mul_op = Mul(root.WithOpName("output"), conv_op, mul_values_op); in TestFoldBatchNormsConv2D()
107 Output conv_op = DepthwiseConv2dNative(root.WithOpName("conv_op"), input_op, in TestFoldBatchNormsDepthwiseConv2dNative() local
115 Output mul_op = Mul(root.WithOpName("output"), conv_op, mul_values_op); in TestFoldBatchNormsDepthwiseConv2dNative()
158 Output conv_op = Conv2D(root.WithOpName("conv_op"), input_op, weights_op, in TestFoldBatchNormsConv2DShared() local
166 Output mul_op = Mul(root.WithOpName("output"), conv_op, mul_values_op); in TestFoldBatchNormsConv2DShared()
174 Mul(root.WithOpName("output_2"), conv_op, mul_values_op_2); in TestFoldBatchNormsConv2DShared()
Dflatten_atrous_test.cc70 Output conv_op = TConvOp(root.WithOpName("conv_op"), space_to_batch_op, in TestFlattenAtrousConv() local
79 root.WithOpName("output"), conv_op, block_shape_op, crops_op); in TestFlattenAtrousConv()
106 EXPECT_EQ(conv_op.node()->type_string(), modified_graph_def.node(2).op()); in TestFlattenAtrousConv()
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_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
474 Output conv_op = Conv2D(root.WithOpName("conv_op"), input_op, weights_op, in TestFoldFusedBatchNormsWithBatchToSpace() local
488 BatchToSpaceND(root.WithOpName("batch_to_space_op"), conv_op, in TestFoldFusedBatchNormsWithBatchToSpace()
Dround_weights_test.cc59 Output conv_op = Conv2D(root.WithOpName("output"), input_op, weights_op, in TestRoundWeights() 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/tools/optimize/
Dsubgraph_quantizer_test.cc77 auto conv_op = subgraph->operators[0].get(); in TEST() local
83 subgraph->tensors[conv_op->inputs[bias_tensor_index]].get(); in TEST()
85 subgraph->tensors[conv_op->inputs[input_tensor_idx]].get(); in TEST()
87 subgraph->tensors[conv_op->inputs[weights_tensor_idx]].get(); in TEST()
89 subgraph->tensors[conv_op->outputs[output_tensor_idx]].get(); in TEST()
176 auto conv_op = subgraph->operators[0].get(); in TEST() local
182 subgraph->tensors[conv_op->inputs[bias_tensor_index]].get(); in TEST()
184 subgraph->tensors[conv_op->inputs[input_tensor_idx]].get(); in TEST()
186 subgraph->tensors[conv_op->inputs[weights_tensor_idx]].get(); in TEST()
188 subgraph->tensors[conv_op->outputs[output_tensor_idx]].get(); in TEST()
Dquantization_utils_test.cc241 auto conv_op = subgraph->operators.at(0).get(); in TEST() local
242 ASSERT_EQ(model.operator_codes.at(conv_op->opcode_index)->builtin_code, in TEST()
244 int32_t weights_tensor_idx = conv_op->inputs[1]; in TEST()
/external/tensorflow/tensorflow/python/kernel_tests/
Dcudnn_determinism_test.py61 conv_op = nn_ops.conv2d(
63 out_shape = conv_op.get_shape()
/external/tensorflow/tensorflow/lite/core/api/
Dflatbuffer_conversions_test.cc108 const Operator* conv_op = in TEST_F() local
115 ParseOpData(conv_op, BuiltinOperator_CONV_2D, &mock_reporter_, in TEST_F()
/external/tensorflow/tensorflow/lite/g3doc/guide/
Dops_version.md178 const auto& conv_op = static_cast<const ConvOperator&>(op);
179 if (conv_op.dilation_width_factor != 1 ||
180 conv_op.dilation_height_factor != 1) {
/external/tensorflow/tensorflow/lite/toco/
Ddump_graphviz.cc373 const auto& conv_op = static_cast<const ConvOperator&>(op); in GetOpAttributes() local
375 AppendF(&stride, "%d", conv_op.stride_width); in GetOpAttributes()
377 AppendF(&stride, "%d", conv_op.stride_height); in GetOpAttributes()
380 (conv_op.padding.type == PaddingType::kSame) ? "same" : "valid"; in GetOpAttributes()
/external/tensorflow/tensorflow/lite/toco/tflite/
Dexport_test.cc349 const TocoOperator& conv_op = in GetVersion() local
351 if (conv_op.dilation_width_factor != 1 || in GetVersion()
352 conv_op.dilation_height_factor != 1) { in GetVersion()
Doperator.cc161 const auto& conv_op = in GetVersion() local
175 if (conv_op.dilation_width_factor != 1 || in GetVersion()
176 conv_op.dilation_height_factor != 1) { in GetVersion()
/external/tensorflow/tensorflow/python/ops/
Dnn_ops.py188 self.conv_op = self._conv1d
199 self.conv_op = conv2d
210 self.conv_op = gen_nn_ops.conv3d
227 return self.conv_op(
1059 self.conv_op = _WithSpaceToBatch(
1078 return self.conv_op(inp, filter)
/external/tensorflow/tensorflow/contrib/quantize/python/
Dquantize_test.py220 conv_op = graph.get_operation_by_name('test/Conv2D')
222 [tensor_in.name for tensor_in in conv_op.inputs])
/external/tensorflow/tensorflow/contrib/rnn/python/ops/
Drnn_cell.py2217 conv_op = nn_ops.conv1d
2220 conv_op = nn_ops.conv2d
2223 conv_op = nn_ops.conv3d
2230 res = conv_op(args[0], kernel, strides, padding="SAME")
2232 res = conv_op(
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc954 miopenFusionOpDescriptor_t conv_op; in SetConvolutionArgs() local
955 auto status = wrap::miopenFusionPlanGetOp(fusion_plan_, op_idx, &conv_op); in SetConvolutionArgs()
961 status = wrap::miopenSetOpArgsConvForward(fusion_args_, conv_op, alpha, in SetConvolutionArgs()
1132 miopenFusionOpDescriptor_t conv_op; in ScopedFusionPlanConvolutionBiasActivation() local
1134 fusion_plan_, &conv_op, conv_descriptor, filter_descriptor); in ScopedFusionPlanConvolutionBiasActivation()
/external/v8/src/compiler/
Dwasm-compiler.cc1672 const Operator* conv_op = ConvertOp(this, opcode); in BuildIntConvertFloat() local
1679 converted_value = graph()->NewNode(conv_op, trunc); in BuildIntConvertFloat()
1681 trunc = graph()->NewNode(conv_op, input); in BuildIntConvertFloat()