| /external/tensorflow/tensorflow/compiler/tests/ |
| D | pooling_ops_test.py | 70 def _VerifyOneTest(self, pool_func, input_sizes, ksize, strides, padding, argument
106 def _VerifyValues(self, pool_func, input_sizes, ksize, strides, padding, argument
294 input_sizes, argument
445 input_sizes, argument
|
| D | depthwise_conv_op_test.py | 457 def _CompareBackpropInput(self, input_sizes, filter_sizes, output_sizes, argument
493 input_sizes, argument
566 def _CompareBackpropInputWithDilation(self, input_sizes, filter_sizes, argument
626 input_sizes, argument
|
| D | pooling_ops_3d_test.py | 42 def _VerifyValues(self, pool_func, input_sizes, window, strides, padding, argument
189 input_sizes, argument
|
| D | conv_node_name_test.py | 37 def _verifyNodeNameMatch(self, layer, input_sizes, filter_sizes, strides, argument
|
| D | conv2d_test.py | 41 input_sizes=None, argument
235 input_sizes=None, argument
549 input_sizes=None, argument
|
| /external/tensorflow/tensorflow/python/kernel_tests/nn_ops/ |
| D | pooling_ops_3d_test.py | 51 def _VerifyOneTest(self, pool_func, input_sizes, window, strides, padding, argument
92 def _VerifyValues(self, pool_func, input_sizes, window, strides, argument
250 input_sizes, argument
|
| D | conv_ops_test.py | 912 input_sizes, argument
961 def _CompareBackpropInput(self, input_sizes, filter_sizes, output_sizes, argument
1135 input_sizes, argument
1180 def _CompareBackFilter(self, input_sizes, filter_sizes, output_sizes, argument
1316 def _RunAndVerifyBackpropInputDilation(self, input_sizes, filter_sizes, argument
1362 def _RunAndVerifyBackpropFilterDilation(self, input_sizes, filter_sizes, argument
1559 input_sizes, argument
1721 input_sizes, argument
|
| D | pooling_ops_test.py | 187 def _VerifyOneType(self, pool_func, input_sizes, ksize, strides, padding, argument
281 def _VerifyOneTest(self, pool_func, input_sizes, ksize, strides, padding, argument
326 input_sizes, argument
1152 input_sizes, argument
1230 input_sizes, argument
1565 expected_input_backprop, input_sizes, output_sizes, argument
|
| D | depthwise_conv_op_base.py | 955 def _CompareBackpropInput(self, input_sizes, filter_sizes, output_sizes, argument
1015 def _CompareBackpropFilter(self, input_sizes, filter_sizes, output_sizes, argument
|
| D | conv_ops_3d_test.py | 786 def _RunAndVerifyBackprop(self, input_sizes, filter_sizes, output_sizes, argument
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | convolution_group_converter.cc | 241 std::vector<int64_t> input_sizes(activation->shape().dimensions().begin(), in HandleBatchGroupCount() local
588 std::vector<int64_t> input_sizes(activation->shape().dimensions().begin(), in HandleConvolution() local
|
| D | dynamic_dimension_inference.cc | 797 HloInstruction* input_sizes = hlo->mutable_operand(0); in HandleDynamicConvolutionInputGrad() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_grad_shape_utils.cc | 174 Status Conv2DBackpropComputeInputShape(const Tensor& input_sizes, in Conv2DBackpropComputeInputShape()
|
| D | conv_grad_ops_3d.cc | 238 const Tensor& input_sizes = context->input(0); in Compute() local
367 const Tensor& input_sizes = context->input(0); in Compute() local
1272 const Tensor& input_sizes = context->input(0); in Compute() local
|
| D | depthwise_conv_grad_op.cc | 614 const Tensor& input_sizes = context->input(0); in Compute() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | conv_ops.cc | 117 xla::XlaOp input_sizes = ctx->Input(0); in Compile() local
|
| D | conv_op_helpers.cc | 296 const xla::PrecisionConfig* precision_config, xla::XlaOp* input_sizes) { in MakeXlaBackpropInputConvOp()
|
| D | strided_slice_op.cc | 455 auto input_sizes = input_shape.dim_sizes(); in CompileAsDynamicUpdateSlice() local
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_conv_ops_test.cc | 157 Node* input_sizes = in DefaultConv2DBwdInput() local
184 Node* input_sizes = in MklConv2DBwdInput() local
|
| /external/mesa3d/src/compiler/nir/ |
| D | nir_opcodes.py | 35 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/grappler/costs/ |
| D | utils_test.cc | 89 NodeDef* input_sizes = graph.add_node(); in TEST() local
|
| /external/eigen/bench/tensors/ |
| D | tensor_benchmarks.h | 493 Eigen::array<TensorIndex, 2> input_sizes; in convolution() local
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | generic_layout_optimizer_test.cc | 130 Output input_sizes = in SimpleConv2DBackpropInput() local
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | nn_ops.py | 2547 input_sizes, argument
3057 input_sizes, argument
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | legalize_hlo.cc | 770 auto input_sizes = rewriter.create<ConstOp>( in matchAndRewrite() local
949 llvm::SmallVector<int> input_sizes = { in MatchResizeOp() local
|