Home
last modified time | relevance | path

Searched defs:input_sizes (Results 1 – 23 of 23) sorted by relevance

/external/tensorflow/tensorflow/compiler/tests/
Dpooling_ops_test.py74 def _VerifyOneTest(self, pool_func, input_sizes, ksize, strides, padding, argument
110 def _VerifyValues(self, pool_func, input_sizes, ksize, strides, padding, argument
298 input_sizes, argument
449 input_sizes, argument
Ddepthwise_conv_op_test.py318 def _CompareBackpropInput(self, input_sizes, filter_sizes, output_sizes, argument
354 input_sizes, argument
Dpooling_ops_3d_test.py46 def _VerifyValues(self, pool_func, input_sizes, window, strides, padding, argument
193 input_sizes, argument
Dconv2d_test.py45 input_sizes=None, argument
239 input_sizes=None, argument
537 input_sizes=None, argument
/external/tensorflow/tensorflow/python/kernel_tests/
Ddepthwise_conv_op_test.py523 def _CompareBackpropInputFloat(self, input_sizes, filter_sizes, output_sizes, argument
543 def _CompareBackpropInputDouble(self, input_sizes, filter_sizes, output_sizes, argument
575 def _CompareBackpropFilterFloat(self, input_sizes, filter_sizes, output_sizes, argument
595 def _CompareBackpropFilterDouble(self, input_sizes, filter_sizes, argument
Dpooling_ops_3d_test.py48 def _VerifyOneTest(self, pool_func, input_sizes, window, strides, padding, argument
89 def _VerifyValues(self, pool_func, input_sizes, window, strides, argument
210 input_sizes, argument
Dconv_ops_test.py704 input_sizes, argument
752 def _CompareBackpropInput(self, input_sizes, filter_sizes, output_sizes, argument
893 input_sizes, argument
938 def _CompareBackFilter(self, input_sizes, filter_sizes, output_sizes, argument
1074 def _RunAndVerifyBackpropInputDilation(self, input_sizes, filter_sizes, argument
1120 def _RunAndVerifyBackpropFilterDilation(self, input_sizes, filter_sizes, argument
1317 input_sizes, argument
1478 input_sizes, argument
Dpooling_ops_test.py108 def _VerifyOneType(self, pool_func, input_sizes, ksize, strides, padding, argument
185 def _VerifyOneTest(self, pool_func, input_sizes, ksize, strides, padding, argument
218 input_sizes, argument
946 input_sizes, argument
1022 input_sizes, argument
1275 expected_input_backprop, input_sizes, output_sizes, argument
Dconv_ops_3d_test.py683 def _RunAndVerifyBackprop(self, input_sizes, filter_sizes, output_sizes, argument
/external/mesa3d/src/compiler/nir/
Dnir_opcodes.py34 def __init__(self, name, output_size, output_type, input_sizes, argument
107 def opcode(name, output_size, output_type, input_sizes, input_types, argument
Dnir.h806 unsigned input_sizes[4]; member
/external/tensorflow/tensorflow/core/kernels/
Dmkl_conv_ops_test.cc176 Node* input_sizes = in DefaultConv2DBwdInput() local
204 Node* input_sizes = in MklConv2DBwdInput() local
Dconv_grad_input_ops.cc314 const Tensor& input_sizes = context->input(0); in Compute() local
633 const Tensor& input_sizes = context->input(0); in Compute() local
Dconv_grad_ops_3d.cc225 const Tensor& input_sizes = context->input(0); in Compute() local
331 const Tensor& input_sizes = context->input(0); in Compute() local
1147 const Tensor& input_sizes = context->input(0); in Compute() local
Ddepthwise_conv_grad_op.cc574 const Tensor& input_sizes = context->input(0); in Compute() local
/external/tensorflow/tensorflow/core/grappler/costs/
Dutils_test.cc89 NodeDef* input_sizes = graph.add_node(); in TEST() local
/external/tensorflow/tensorflow/contrib/hvx/hexagon_controller/src_soc_interface/
Dsoc_interface.c92 bool soc_interface_AllocateInOutNodeBuffers(int input_count, int* input_sizes, in soc_interface_AllocateInOutNodeBuffers()
/external/eigen/bench/tensors/
Dtensor_benchmarks.h418 Eigen::array<TensorIndex, 2> input_sizes; in convolution() local
/external/tensorflow/tensorflow/core/kernels/hexagon/
Dhexagon_control_wrapper.cc101 std::vector<int> input_sizes; in Init() local
/external/tensorflow/tensorflow/contrib/hvx/hexagon_controller/src_impl/
Dhexagon_controller.c281 int* input_sizes, in hexagon_controller_AllocateMultipleNodeDataBuffers()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dlayout_optimizer_test.cc135 Output input_sizes = in SimpleConv2DBackpropInput() local
241 auto input_sizes = GetAttrValue(*input_sizes_node); in TEST_F() local
/external/tensorflow/tensorflow/python/ops/
Dnn_ops.py2016 input_sizes, argument
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes_test.cc3119 auto input_sizes = in TEST_F() local