Home
last modified time | relevance | path

Searched defs:tensor_in_sizes (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/nn_ops/
H A Dconv_ops_test.py182 def _SetupValuesForDevice(self, tensor_in_sizes, filter_in_sizes, dilations, argument
229 def _CompareFwdValues(self, tensor_in_sizes, filter_in_sizes, conv_strides, argument
265 def _ComputeReferenceDilatedConv(self, tensor_in_sizes, filter_in_sizes, argument
303 def _VerifyDilatedConvValues(self, tensor_in_sizes, filter_in_sizes, strides, argument
323 tensor_in_sizes, argument
374 tensor_in_sizes, argument
791 def _VerifyGroupConvFwd(self, tensor_in_sizes, filter_in_sizes, dilations, argument
2674 def _VerifyValues(self, tensor_in_sizes, filter_in_sizes, stride, padding, argument
2785 tensor_in_sizes, argument
2953 def _CompareFwdConv2D(self, tensor_in_sizes, filter_in_sizes, conv_strides, argument
H A Dconv_ops_3d_test.py68 def _SetupValuesForDevice(self, tensor_in_sizes, filter_in_sizes, stride, argument
97 def _VerifyValues(self, tensor_in_sizes, filter_in_sizes, stride, padding, argument
123 def _ComputeReferenceDilatedConv(self, tensor_in_sizes, filter_in_sizes, argument
166 def _VerifyDilatedConvValues(self, tensor_in_sizes, filter_in_sizes, stride, argument
H A Ddepthwise_conv_op_base.py300 tensor_in_sizes, argument
530 def _VerifyHandValues(self, tensor_in_sizes, filter_in_sizes, stride, padding, argument
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tests/
H A Ddepthwise_conv_op_test.py148 tensor_in_sizes, argument
252 def _VerifyHandValues(self, tensor_in_sizes, filter_in_sizes, stride, padding, argument
349 tensor_in_sizes, argument
/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/image_ops/
H A Dattention_ops_test.py30 def _VerifyValues(self, tensor_in_sizes, glimpse_sizes, offsets, argument
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/
H A Dquantized_conv_ops_test.py32 def _VerifyValues(self, tensor_in_sizes, filter_in_sizes, stride, padding, argument