Home
last modified time | relevance | path

Searched defs:filter_tensor (Results 1 – 16 of 16) sorted by relevance

/external/tensorflow/tensorflow/lite/tools/versioning/
Dop_version.cc688 const Tensor* filter_tensor = in GetOpSignature() local
799 const Tensor* filter_tensor = in GetOpSignature() local
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dmkl_remapper_test.cc108 auto filter_tensor = GenerateRandomTensor<DT_FLOAT>( in FuseConv2DWithBiasAndAddNOrAdd() local
478 auto filter_tensor = GenerateRandomTensor<DT_FLOAT>( in TEST_F() local
/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_conv_grad_input_ops.cc316 const Tensor& filter_tensor = MklGetInput(context, kFilterIdx); in Compute() local
535 const Tensor& filter_tensor) { in MakeFilterTfShape()
Dmkl_conv_ops.cc501 const Tensor& filter_tensor = MklGetInput(context, kInputIndex_Filter); in Compute() local
976 Tensor** filter_tensor, in AllocatePersistentTensor()
1005 Tensor** filter_tensor) { in AllocatePersistentTensor()
1012 Tensor** filter_tensor) { in AllocateFilterOutputTensor()
1092 Tfilter* filter_data, const Tensor& filter_tensor, in CacheFilter()
Dmkl_conv_grad_filter_ops.cc373 const Tensor& filter_tensor = MklGetInput(context, kFilterIdx); in Compute() local
639 const Tensor& filter_tensor) { in MakeFilterTfShape()
/external/tensorflow/tensorflow/lite/micro/kernels/
Dconv_test.cc468 TfLiteTensor filter_tensor = tflite::testing::CreateQuantizedTensor( in TF_LITE_MICRO_TEST() local
553 TfLiteTensor filter_tensor = in TF_LITE_MICRO_TEST() local
680 TfLiteTensor filter_tensor = tflite::testing::CreateQuantizedTensor( in TF_LITE_MICRO_TEST() local
Ddepthwise_conv_test.cc187 TfLiteTensor filter_tensor = CreateSymmetricPerChannelQuantizedTensor( in TestDepthwiseConvQuantizedPerChannel() local
720 TfLiteTensor filter_tensor = tflite::testing::CreateQuantizedTensor( in TF_LITE_MICRO_TEST() local
815 TfLiteTensor filter_tensor = in TF_LITE_MICRO_TEST() local
949 TfLiteTensor filter_tensor = tflite::testing::CreateQuantizedTensor( in TF_LITE_MICRO_TEST() local
Dtranspose_conv_test.cc150 TfLiteTensor filter_tensor = CreateSymmetricPerChannelQuantizedTensor( in TestTransposeConvQuantized() local
Dconv_test_common.cc206 TfLiteTensor filter_tensor = CreateSymmetricPerChannelQuantizedTensor( in TestConvQuantizedPerChannel() local
/external/tensorflow/tensorflow/compiler/tests/
Ddepthwise_conv_op_test.py35 def ReferenceDepthwiseConv2D(input_tensor, filter_tensor, strides, padding, argument
/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/tests/
Dconv_test.cc99 auto* filter_tensor = interpreter_->tensor(filter_); in QuantizedConvolutionOpModel() local
/external/tensorflow/tensorflow/lite/delegates/xnnpack/
Dxnnpack_delegate.cc1242 const TfLiteTensor& filter_tensor = tensors[node->inputs->data[1]]; in VisitConv2DNode() local
1337 const TfLiteTensor& filter_tensor = tensors[node->inputs->data[1]]; in VisitDepthwiseConv2DNode() local
1570 const TfLiteTensor& filter_tensor = tensors[node->inputs->data[1]]; in VisitFullyConnectedNode() local
2034 const TfLiteTensor& filter_tensor = tensors[node->inputs->data[1]]; in VisitMediaPipeDeconvolutionNode() local
/external/tensorflow/tensorflow/lite/micro/kernels/arc_mli/
Dconv_slicing_test.cc214 TfLiteTensor filter_tensor = CreateSymmetricPerChannelQuantizedTensor( in TestConvQuantizedPerChannel() local
Ddepthwise_conv_slicing_test.cc143 TfLiteTensor filter_tensor = CreateSymmetricPerChannelQuantizedTensor( in TestDepthwiseConvQuantizedPerChannel() local
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate.cc1794 const auto& filter_tensor = context->tensors[node->inputs->data[1]]; in Validate() local
3144 const TfLiteTensor& filter_tensor = in Map() local
/external/tensorflow/tensorflow/tools/graph_transforms/
Dquantize_nodes_test.cc329 Tensor filter_tensor( in TestQuantizeConv2D() local