Home
last modified time | relevance | path

Searched refs:filter_tensor (Results 1 – 22 of 22) sorted by relevance

/external/tensorflow/tensorflow/lite/testing/op_tests/
Dconv2d_transpose.py53 filter_tensor = tf.compat.v1.placeholder(
56 input_tensors = [input_tensor, filter_tensor]
66 filter_tensor,
Dconv_to_depthwiseconv_with_shared_weights.py56 filter_tensor = tf.constant(
64 filter_tensor,
71 filter_tensor,
Dconv_with_shared_weights.py58 filter_tensor = tf.constant(
69 filter_tensor,
76 filter_tensor,
/external/tensorflow/tensorflow/lite/micro/kernels/
Ddepthwise_conv_test.cc187 TfLiteTensor filter_tensor = CreateSymmetricPerChannelQuantizedTensor( in TestDepthwiseConvQuantizedPerChannel() local
218 filter_tensor, in TestDepthwiseConvQuantizedPerChannel()
720 TfLiteTensor filter_tensor = tflite::testing::CreateQuantizedTensor( in TF_LITE_MICRO_TEST() local
727 filter_tensor.quantization = {kTfLiteAffineQuantization, &filter_quant}; in TF_LITE_MICRO_TEST()
757 filter_tensor, in TF_LITE_MICRO_TEST()
815 TfLiteTensor filter_tensor = in TF_LITE_MICRO_TEST() local
838 filter_tensor, in TF_LITE_MICRO_TEST()
850 filter_tensor.quantization.params); in TF_LITE_MICRO_TEST()
949 TfLiteTensor filter_tensor = tflite::testing::CreateQuantizedTensor( in TF_LITE_MICRO_TEST() local
958 filter_tensor.quantization = {kTfLiteAffineQuantization, &filter_quant}; in TF_LITE_MICRO_TEST()
[all …]
Dconv_test.cc468 TfLiteTensor filter_tensor = tflite::testing::CreateQuantizedTensor( in TF_LITE_MICRO_TEST() local
476 filter_tensor.quantization = {kTfLiteAffineQuantization, &filter_quant}; in TF_LITE_MICRO_TEST()
507 filter_tensor, in TF_LITE_MICRO_TEST()
553 TfLiteTensor filter_tensor = in TF_LITE_MICRO_TEST() local
577 filter_tensor, in TF_LITE_MICRO_TEST()
587 filter_tensor.quantization.params); in TF_LITE_MICRO_TEST()
680 TfLiteTensor filter_tensor = tflite::testing::CreateQuantizedTensor( in TF_LITE_MICRO_TEST() local
689 filter_tensor.quantization = {kTfLiteAffineQuantization, &filter_quant}; in TF_LITE_MICRO_TEST()
725 filter_tensor, in TF_LITE_MICRO_TEST()
Dconv_test_common.cc206 TfLiteTensor filter_tensor = CreateSymmetricPerChannelQuantizedTensor( in TestConvQuantizedPerChannel() local
234 filter_tensor, in TestConvQuantizedPerChannel()
Dtranspose_conv_test.cc150 TfLiteTensor filter_tensor = CreateSymmetricPerChannelQuantizedTensor( in TestTransposeConvQuantized() local
164 CreateTensor(output_shape, output_shape_dims), filter_tensor, in TestTransposeConvQuantized()
/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_conv_grad_input_ops.cc316 const Tensor& filter_tensor = MklGetInput(context, kFilterIdx); in Compute() local
334 src_tensor, filter_tensor.shape(), in Compute()
341 TensorShape filter_tf_shape = MakeFilterTfShape(context, filter_tensor); in Compute()
463 filter.SetUsrMem(fwd_filter_md, &filter_tensor); in Compute()
469 static_cast<T*>(const_cast<T*>(filter_tensor.flat<T>().data())); in Compute()
535 const Tensor& filter_tensor) { in MakeFilterTfShape() argument
Dmkl_conv_ops.cc501 const Tensor& filter_tensor = MklGetInput(context, kInputIndex_Filter); in Compute() local
629 filter.SetUsrMem(filter_md, &filter_tensor); in Compute()
700 CacheFilter(context, conv_fwd_pd, filter_data, filter_tensor, in Compute()
707 filter.SetUsrMem(filter_md, &filter_tensor); in Compute()
722 const_cast<Tfilter*>(filter_tensor.flat<Tfilter>().data())); in Compute()
976 Tensor** filter_tensor, in AllocatePersistentTensor() argument
978 DCHECK(filter_tensor); in AllocatePersistentTensor()
984 &cached_filter_data_ptensor_, filter_tensor)); in AllocatePersistentTensor()
1005 Tensor** filter_tensor) { in AllocatePersistentTensor() argument
1006 AllocatePersistentTensor(context, conv_prim_desc, filter_tensor, nullptr); in AllocatePersistentTensor()
[all …]
Dmkl_conv_grad_filter_ops.cc373 const Tensor& filter_tensor = MklGetInput(context, kFilterIdx); in Compute() local
389 TensorShape filter_tf_shape = MakeFilterTfShape(context, filter_tensor); in Compute()
639 const Tensor& filter_tensor) { in MakeFilterTfShape() argument
641 CHECK_EQ(TensorShapeUtils::IsVector(filter_tensor.shape()), true); in MakeFilterTfShape()
642 CHECK_EQ(TensorShapeUtils::MakeShape(filter_tensor.vec<int32>(), in MakeFilterTfShape()
/external/tensorflow/tensorflow/lite/tools/versioning/
Dop_version.cc688 const Tensor* filter_tensor = in GetOpSignature() local
691 filter_tensor->quantization(); in GetOpSignature()
692 int num_channels = filter_tensor->shape()->Get(3); in GetOpSignature()
799 const Tensor* filter_tensor = in GetOpSignature() local
802 filter_tensor->quantization(); in GetOpSignature()
803 int num_channels = filter_tensor->shape()->Get(0); in GetOpSignature()
/external/tensorflow/tensorflow/lite/delegates/xnnpack/
Dxnnpack_delegate.cc1242 const TfLiteTensor& filter_tensor = tensors[node->inputs->data[1]]; in VisitConv2DNode() local
1244 logging_context, filter_tensor, node->inputs->data[1], node_index)); in VisitConv2DNode()
1245 TF_LITE_ENSURE_STATUS(CheckTensorShape(logging_context, filter_tensor, 4, in VisitConv2DNode()
1249 logging_context, filter_tensor, node->inputs->data[1], node_index)); in VisitConv2DNode()
1277 const int output_channels = filter_tensor.dims->data[0]; in VisitConv2DNode()
1278 const int kernel_height = filter_tensor.dims->data[1]; in VisitConv2DNode()
1279 const int kernel_width = filter_tensor.dims->data[2]; in VisitConv2DNode()
1280 const int input_channels = filter_tensor.dims->data[3]; in VisitConv2DNode()
1337 const TfLiteTensor& filter_tensor = tensors[node->inputs->data[1]]; in VisitDepthwiseConv2DNode() local
1339 logging_context, filter_tensor, node->inputs->data[1], node_index)); in VisitDepthwiseConv2DNode()
[all …]
/external/tensorflow/tensorflow/lite/micro/kernels/arc_mli/
Dconv_slicing_test.cc214 TfLiteTensor filter_tensor = CreateSymmetricPerChannelQuantizedTensor( in TestConvQuantizedPerChannel() local
224 (TfLiteAffineQuantization*)filter_tensor.quantization.params; in TestConvQuantizedPerChannel()
227 int filter_size = filter_tensor.bytes; in TestConvQuantizedPerChannel()
229 filter_tensor.data.int8[i] = filter_data[i]; in TestConvQuantizedPerChannel()
258 filter_tensor, in TestConvQuantizedPerChannel()
Ddepthwise_conv_slicing_test.cc143 TfLiteTensor filter_tensor = CreateSymmetricPerChannelQuantizedTensor( in TestDepthwiseConvQuantizedPerChannel() local
173 filter_tensor, in TestDepthwiseConvQuantizedPerChannel()
/external/tensorflow/tensorflow/python/kernel_tests/
Dconv_ops_3d_test.py399 filter_tensor = constant_op.constant(
403 filter_tensor,
515 filter_tensor = constant_op.constant(
527 filter_tensor,
543 filter_tensor, filter_shape, conv, output_shape)
Ddepthwise_conv_op_test.py619 filter_tensor = constant_op.constant(
647 filter_tensor,
662 filter_tensor, filter_shape, depthwise_conv2d, output_shape)
Dconv_ops_test.py402 filter_tensor = self._CreateNumpyTensor(filter_in_sizes)
407 filter_tensor, [1] + list(strides) + [1],
1899 filter_tensor = constant_op.constant(
1912 filter_tensor,
1926 jacob_t, jacob_n = gradient_checker.compute_gradient(filter_tensor,
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dmkl_remapper_test.cc108 auto filter_tensor = GenerateRandomTensor<DT_FLOAT>( in FuseConv2DWithBiasAndAddNOrAdd() local
116 {"filter", filter_tensor}, in FuseConv2DWithBiasAndAddNOrAdd()
478 auto filter_tensor = GenerateRandomTensor<DT_FLOAT>( in TEST_F() local
486 {"filter", filter_tensor}, in TEST_F()
/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/tests/
Dconv_test.cc99 auto* filter_tensor = interpreter_->tensor(filter_); in QuantizedConvolutionOpModel() local
100 filter_tensor->allocation_type = kTfLiteMmapRo; in QuantizedConvolutionOpModel()
/external/tensorflow/tensorflow/compiler/tests/
Ddepthwise_conv_op_test.py35 def ReferenceDepthwiseConv2D(input_tensor, filter_tensor, strides, padding, argument
40 in_channels = filter_tensor.shape[2]
50 filter_slice = filter_tensor[:, :, channel:channel+1, :]
/external/tensorflow/tensorflow/tools/graph_transforms/
Dquantize_nodes_test.cc329 Tensor filter_tensor( in TestQuantizeConv2D() local
331 test::FillValues<float>(&filter_tensor, filter_values); in TestQuantizeConv2D()
333 Const(root.WithOpName("filter_op"), Input::Initializer(filter_tensor)); in TestQuantizeConv2D()
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate.cc1794 const auto& filter_tensor = context->tensors[node->inputs->data[1]]; in Validate() local
1795 if (filter_tensor.quantization.type == kTfLiteAffineQuantization) { in Validate()
1798 filter_tensor.quantization.params); in Validate()
3144 const TfLiteTensor& filter_tensor = in Map() local
3150 input_tensor.params.scale * filter_tensor.params.scale; in Map()
3153 input_tensor.params.scale * filter_tensor.params.scale, in Map()