Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/compiler/tests/
Ddepthwise_conv_op_test.py457 def _CompareBackpropInput(self, input_sizes, filter_sizes, output_sizes, argument
493 input_sizes, argument
562 def _CompareBackpropInputWithDilation(self, input_sizes, filter_sizes, argument
622 input_sizes, argument
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
Dconv_node_name_test.py41 def _verifyNodeNameMatch(self, layer, input_sizes, filter_sizes, strides, 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.py624 def _CompareBackpropInputFloat(self, input_sizes, filter_sizes, output_sizes, argument
644 def _CompareBackpropInputDouble(self, input_sizes, filter_sizes, output_sizes, argument
680 def _CompareBackpropFilterFloat(self, input_sizes, filter_sizes, output_sizes, argument
700 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
230 input_sizes, argument
Dpooling_ops_test.py109 def _VerifyOneType(self, pool_func, input_sizes, ksize, strides, padding, argument
186 def _VerifyOneTest(self, pool_func, input_sizes, ksize, strides, padding, argument
221 input_sizes, argument
969 input_sizes, argument
1045 input_sizes, argument
1298 expected_input_backprop, input_sizes, output_sizes, argument
Dconv_ops_test.py823 input_sizes, argument
871 def _CompareBackpropInput(self, input_sizes, filter_sizes, output_sizes, argument
1012 input_sizes, argument
1057 def _CompareBackFilter(self, input_sizes, filter_sizes, output_sizes, argument
1193 def _RunAndVerifyBackpropInputDilation(self, input_sizes, filter_sizes, argument
1239 def _RunAndVerifyBackpropFilterDilation(self, input_sizes, filter_sizes, argument
1436 input_sizes, argument
1598 input_sizes, argument
Dconv_ops_3d_test.py720 def _RunAndVerifyBackprop(self, input_sizes, filter_sizes, output_sizes, argument
/external/tensorflow/tensorflow/compiler/xla/service/
Dconvolution_group_converter.cc234 std::vector<int64> input_sizes(activation->shape().dimensions().begin(), in HandleBatchGroupCount() local
577 std::vector<int64> input_sizes(activation->shape().dimensions().begin(), in HandleConvolution() local
/external/mesa3d/src/compiler/nir/
Dnir_opcodes.py35 def __init__(self, name, output_size, output_type, input_sizes, argument
153 def opcode(name, output_size, output_type, input_sizes, input_types, argument
/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_ops_3d.cc233 const Tensor& input_sizes = context->input(0); in Compute() local
340 const Tensor& input_sizes = context->input(0); in Compute() local
1158 const Tensor& input_sizes = context->input(0); in Compute() local
Dconv_grad_input_ops.cc450 const Tensor& input_sizes = context->input(0); in Compute() local
549 const Tensor& input_sizes = context->input(0); in Compute() local
Ddepthwise_conv_grad_op.cc587 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/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/core/grappler/optimizers/
Dgeneric_layout_optimizer_test.cc91 Output input_sizes = in SimpleConv2DBackpropInput() local
Dlayout_optimizer_test.cc135 Output input_sizes = in SimpleConv2DBackpropInput() local
241 auto input_sizes = GetAttrValue(*input_sizes_node); in TEST_F() local
Dgeneric_layout_optimizer_transposer_test.cc233 auto input_sizes = ops::Const(scope.WithOpName("input_sizes"), in CreateSimpleConv2DBackpropInput() local
/external/tensorflow/tensorflow/python/ops/
Dnn_ops.py2079 input_sizes, argument
/external/mesa3d/src/broadcom/compiler/
Dv3d_compiler.h781 uint8_t input_sizes[V3D_MAX_GS_INPUTS / 4]; member
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes_test.cc3720 auto input_sizes = in TEST_F() local
4009 auto input_sizes = in TEST_F() local

12