Home
last modified time | relevance | path

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

12

/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.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
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
553 input_sizes=None, argument
/external/tensorflow/tensorflow/python/kernel_tests/
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
Dconv_ops_test.py900 input_sizes, argument
949 def _CompareBackpropInput(self, input_sizes, filter_sizes, output_sizes, argument
1106 input_sizes, argument
1151 def _CompareBackFilter(self, input_sizes, filter_sizes, output_sizes, argument
1287 def _RunAndVerifyBackpropInputDilation(self, input_sizes, filter_sizes, argument
1333 def _RunAndVerifyBackpropFilterDilation(self, input_sizes, filter_sizes, argument
1530 input_sizes, argument
1692 input_sizes, argument
Dpooling_ops_test.py118 def _VerifyOneType(self, pool_func, input_sizes, ksize, strides, padding, argument
201 def _VerifyOneTest(self, pool_func, input_sizes, ksize, strides, padding, argument
246 input_sizes, argument
1110 input_sizes, argument
1188 input_sizes, argument
1523 expected_input_backprop, input_sizes, output_sizes, argument
Ddepthwise_conv_op_test.py900 def _CompareBackpropInput(self, input_sizes, filter_sizes, output_sizes, argument
958 def _CompareBackpropFilter(self, input_sizes, filter_sizes, output_sizes, argument
Dconv_ops_3d_test.py779 def _RunAndVerifyBackprop(self, input_sizes, filter_sizes, output_sizes, argument
/external/tensorflow/tensorflow/compiler/xla/service/
Dconvolution_group_converter.cc236 std::vector<int64> input_sizes(activation->shape().dimensions().begin(), in HandleBatchGroupCount() local
582 std::vector<int64> input_sizes(activation->shape().dimensions().begin(), in HandleConvolution() local
/external/tensorflow/tensorflow/core/kernels/
Dconv_grad_shape_utils.cc169 Status Conv2DBackpropComputeInputShape(const Tensor& input_sizes, in Conv2DBackpropComputeInputShape()
Dconv_grad_ops_3d.cc234 const Tensor& input_sizes = context->input(0); in Compute() local
341 const Tensor& input_sizes = context->input(0); in Compute() local
1160 const Tensor& input_sizes = context->input(0); in Compute() local
Ddepthwise_conv_grad_op.cc609 const Tensor& input_sizes = context->input(0); in Compute() local
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dconv_ops.cc115 xla::XlaOp input_sizes = ctx->Input(0); in Compile() local
Dconv_op_helpers.cc296 const xla::PrecisionConfig* precision_config, xla::XlaOp* input_sizes) { in MakeXlaBackpropInputConvOp()
Dstrided_slice_op.cc418 auto input_sizes = input_shape.dim_sizes(); in CompileAsDynamicUpdateSlice() 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/mkl/
Dmkl_conv_ops_test.cc162 Node* input_sizes = in DefaultConv2DBwdInput() local
190 Node* input_sizes = in MklConv2DBwdInput() 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.cc130 Output input_sizes = in SimpleConv2DBackpropInput() local
Dlayout_optimizer_test.cc141 Output input_sizes = in SimpleConv2DBackpropInput() local
247 auto input_sizes = GetAttrValue(*input_sizes_node); in TEST_F() local
/external/tensorflow/tensorflow/python/ops/
Dnn_ops.py2358 input_sizes, argument
2860 input_sizes, argument

12