Home
last modified time | relevance | path

Searched refs:conv_output (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/common/
Dgpu_model_test_util.cc59 tflite::gpu::Value* conv_output = nullptr; in TestLinkingConvolutionAndCosOp() local
60 RETURN_IF_ERROR(ConnectTwoNodes(&graph, conv_node, cos_node, &conv_output)); in TestLinkingConvolutionAndCosOp()
61 conv_output->tensor.type = DataType::FLOAT32; in TestLinkingConvolutionAndCosOp()
62 conv_output->tensor.shape = BHWC(1, 32, 32, 16); in TestLinkingConvolutionAndCosOp()
108 conv_output->tensor.shape, &intermediate)); in TestLinkingConvolutionAndCosOp()
162 tflite::gpu::Value* conv_output = nullptr; in TestLinkingConvolution2InputMul2InputMul() local
163 RETURN_IF_ERROR(ConnectTwoNodes(&graph, conv_node, mul0_node, &conv_output)); in TestLinkingConvolution2InputMul2InputMul()
165 conv_output->tensor.type = DataType::FLOAT32; in TestLinkingConvolution2InputMul2InputMul()
166 conv_output->tensor.shape = BHWC(1, 32, 32, 16); in TestLinkingConvolution2InputMul2InputMul()
233 conv_output->tensor.shape, &intermediate0)); in TestLinkingConvolution2InputMul2InputMul()
[all …]
/external/tensorflow/tensorflow/compiler/mlir/quantization/tensorflow/passes/
Dreplace_cast_hacks_with_tf_xla_ops.cc97 Value filter, Value input_zp, Value conv_output, in CreateXLAConvOp() argument
144 loc, /*output_type=*/conv_output.getType(), in CreateXLAConvOp()
163 Value conv_output, ArrayAttr strides, in CreateXLAConvOpFromTFConv2DOp() argument
177 return CreateXLAConvOp(builder, loc, input, filter, input_zp, conv_output, in CreateXLAConvOpFromTFConv2DOp()
185 Value conv_output, ArrayAttr strides, ArrayAttr dilations, in CreateXLAConvOpFromTFDepthwiseConv2DOp() argument
204 return CreateXLAConvOp(builder, loc, input, new_filter, input_zp, conv_output, in CreateXLAConvOpFromTFDepthwiseConv2DOp()
/external/autotest/client/bin/
Dpackage.py277 conv_output = utils.system_output('alien --to-deb %s 2>/dev/null'
279 converted_package = re.findall(deb_pattern, conv_output)[0]
282 conv_output = utils.system_output('alien --to-rpm %s 2>/dev/null'
284 converted_package = re.findall(rpm_pattern, conv_output)[0]
/external/executorch/exir/backend/test/
Dbackend_with_delegate_mapping_demo.py199 conv_output = self.conv_layer(x)
202 tan_output = torch.tan(conv_output)
/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/
Dconv_2d_builder.cc212 TensorID conv_output = conv_op->AddOutput( in BuildDilatedDwConv() local
222 batch_to_space_op->AddInput(conv_output); in BuildDilatedDwConv()
/external/pytorch/test/quantization/jit/
Dtest_ondevice_quantization.py45 conv_output = self.conv(x)
46 y = self.mymod(conv_output)
/external/tensorflow/tensorflow/lite/toco/
Dexport_tensorflow.cc396 std::string conv_output = src_op.outputs[0]; in ConvertConvOperator() local
398 conv_output += "/conv"; in ConvertConvOperator()
403 conv2d_op->set_name(conv_output); in ConvertConvOperator()
440 biasadd_op->add_input(conv_output); in ConvertConvOperator()
464 std::string conv_output = src_op.outputs[0]; in ConvertDepthwiseConvOperator() local
466 conv_output += "/conv"; in ConvertDepthwiseConvOperator()
471 dc2d_op->set_name(conv_output); in ConvertDepthwiseConvOperator()
531 biasadd_op->add_input(conv_output); in ConvertDepthwiseConvOperator()
/external/tensorflow/tensorflow/compiler/mlir/quantization/tensorflow/tests/
Dreplace_cast_hacks_with_tf_xla_ops.mlir220 // CHECK: %[[conv_output:.*]] = "tf.XlaConvV2"(%[[input_padded]], %[[filter]], {{.*}}, {{.*}}, {{.*…
221 // CHECK: %[[conv_output_sub:.*]] = "tf.Sub"(%[[conv_output]], {{.*}}) : (tensor<?x?x?x2xi32>, tens…
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Doptimize_patterns.td192 (BinaryOp (TFL_Conv2DOp:$conv_output $input,
207 (HasOneUse $conv_output)]>;
/external/pytorch/test/jit/
Dtest_freezing.py3440 conv_output = self.conv(x)
3441 return conv_output, self.conv(torch.add(x, x))