Searched defs:filter_tensor (Results 1 – 12 of 12) sorted by relevance
| /external/tensorflow/tensorflow/lite/tools/versioning/ |
| D | op_signature.cc | 141 const Tensor* filter_tensor = in GetOpSignature() local
189 const Tensor* filter_tensor = in GetOpSignature() local
|
| /external/tensorflow/tensorflow/lite/python/ |
| D | lite_flex_test.py | 401 def conv_func(self, in_tensor, filter_tensor): argument
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_conv_grad_input_ops.cc | 309 const Tensor& filter_tensor = MklGetInput(context, kFilterIdx); in Compute() local
538 const Tensor& filter_tensor) { in MakeFilterTfShape()
|
| D | mkl_conv_grad_filter_ops.cc | 367 const Tensor& filter_tensor = MklGetInput(context, kFilterIdx); in Compute() local
668 const Tensor& filter_tensor) { in MakeFilterTfShape()
|
| D | mkl_conv_ops.cc | 629 const Tensor& filter_tensor = MklGetInput(context, kInputIndex_Filter); in Compute() local
1221 Tensor** filter_tensor, in AllocateTensor()
1250 Tensor** filter_tensor) { in AllocateTensor()
1257 Tensor** filter_tensor) { in AllocateFilterOutputTensor()
1296 Tfilter* filter_data, const Tensor& filter_tensor, in CacheFilter()
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | depthwise_conv_op_test.py | 31 def ReferenceDepthwiseConv2D(input_tensor, filter_tensor, strides, padding, argument
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | mkl_remapper_test.cc | 108 auto filter_tensor = GenerateRandomTensor<DT_FLOAT>( in FuseConv2DWithBiasAndAddNOrAdd() local
484 auto filter_tensor = GenerateRandomTensor<DT_FLOAT>( in TEST_F() local
|
| D | remapper_test.cc | 2176 auto filter_tensor = GenerateRandomTensor<DT_FLOAT>( in TEST_F() local
|
| /external/tensorflow/tensorflow/lite/delegates/hexagon/builders/tests/ |
| D | conv_test.cc | 101 auto* filter_tensor = interpreter_->tensor(filter_); in QuantizedConvolutionOpModel() local
|
| /external/tensorflow/tensorflow/lite/delegates/xnnpack/ |
| D | xnnpack_delegate.cc | 2567 const TfLiteTensor& filter_tensor = tensors[node->inputs->data[1]]; in VisitConv2DNode() local
2676 const TfLiteTensor& filter_tensor = tensors[node->inputs->data[1]]; in VisitDepthwiseConv2DNode() local
2964 const TfLiteTensor& filter_tensor = tensors[node->inputs->data[1]]; in VisitFullyConnectedNode() local
3522 const TfLiteTensor& filter_tensor = tensors[node->inputs->data[1]]; in VisitMediaPipeDeconvolutionNode() local
4604 const TfLiteTensor& filter_tensor = tensors[filter_tensor_index]; in VisitTransposeConvNode() local
|
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | nnapi_delegate.cc | 2410 const auto& filter_tensor = context->tensors[node->inputs->data[1]]; in Validate() local
3598 const TfLiteTensor& filter_tensor = in Map() local
3650 const auto& filter_tensor = context->tensors[filter_id]; in Map() local
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | quantize_nodes_test.cc | 329 Tensor filter_tensor( in TestQuantizeConv2D() local
|