Searched defs:filter_tensor (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/lite/tools/versioning/ |
| D | op_signature.cc | 129 const Tensor* filter_tensor = in GetOpSignature() local
175 const Tensor* filter_tensor = in GetOpSignature() local
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | mkl_remapper_test.cc | 108 auto filter_tensor = GenerateRandomTensor<DT_FLOAT>( in FuseConv2DWithBiasAndAddNOrAdd() local
477 auto filter_tensor = GenerateRandomTensor<DT_FLOAT>( in TEST_F() local
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_conv_grad_input_ops.cc | 316 const Tensor& filter_tensor = MklGetInput(context, kFilterIdx); in Compute() local
539 const Tensor& filter_tensor) { in MakeFilterTfShape()
|
| D | mkl_conv_grad_filter_ops.cc | 373 const Tensor& filter_tensor = MklGetInput(context, kFilterIdx); in Compute() local
666 const Tensor& filter_tensor) { in MakeFilterTfShape()
|
| D | mkl_conv_ops.cc | 614 const Tensor& filter_tensor = MklGetInput(context, kInputIndex_Filter); in Compute() local
1167 Tensor** filter_tensor, in AllocateTensor()
1195 Tensor** filter_tensor) { in AllocateTensor()
1202 Tensor** filter_tensor) { in AllocateFilterOutputTensor()
1281 Tfilter* filter_data, const Tensor& filter_tensor, in CacheFilter()
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | depthwise_conv_op_test.py | 35 def ReferenceDepthwiseConv2D(input_tensor, filter_tensor, strides, padding, argument
|
| /external/tensorflow/tensorflow/lite/delegates/hexagon/builders/tests/ |
| D | conv_test.cc | 99 auto* filter_tensor = interpreter_->tensor(filter_); in QuantizedConvolutionOpModel() local
|
| /external/tensorflow/tensorflow/lite/delegates/xnnpack/ |
| D | xnnpack_delegate.cc | 1739 const TfLiteTensor& filter_tensor = tensors[node->inputs->data[1]]; in VisitConv2DNode() local
1843 const TfLiteTensor& filter_tensor = tensors[node->inputs->data[1]]; in VisitDepthwiseConv2DNode() local
2086 const TfLiteTensor& filter_tensor = tensors[node->inputs->data[1]]; in VisitFullyConnectedNode() local
2559 const TfLiteTensor& filter_tensor = tensors[node->inputs->data[1]]; in VisitMediaPipeDeconvolutionNode() local
|
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | nnapi_delegate.cc | 2001 const auto& filter_tensor = context->tensors[node->inputs->data[1]]; in Validate() local
3038 const TfLiteTensor& filter_tensor = in Map() local
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | quantize_nodes_test.cc | 329 Tensor filter_tensor( in TestQuantizeConv2D() local
|