| /external/pytorch/aten/src/ATen/native/vulkan/glsl/ |
| D | texel_access.h | 9 ivec4 input_sizes) { in map_output_pos_to_input_pos()
28 ivec4 input_sizes, in load_texel()
|
| /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 | conv_node_name_test.py | 37 def _verifyNodeNameMatch(self, layer, input_sizes, filter_sizes, strides, argument
|
| D | pooling_ops_3d_test.py | 42 def _VerifyValues(self, pool_func, input_sizes, window, strides, padding, argument
189 input_sizes, argument
|
| D | conv2d_test.py | 41 input_sizes=None, argument
235 input_sizes=None, argument
549 input_sizes=None, argument
|
| /external/pytorch/torch/csrc/distributed/c10d/quantization/ |
| D | quantization.cpp | 49 const auto input_sizes = input.sizes(); in _float_to_bfloat16_cpu() local
68 const auto input_sizes = input.sizes(); in _bfloat16_to_float_cpu() local
|
| /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
|
| /external/pytorch/aten/src/ATen/native/ |
| D | UnfoldBackward.cpp | 19 IntArrayRef input_sizes, in unfold_backward()
|
| D | LinearAlgebraUtils.h | 583 IntArrayRef input_sizes = input.sizes(); in is_blas_compatible_column_major_order() local
604 IntArrayRef input_sizes = input.sizes(); in is_blas_compatible_row_major_order() local
|
| /external/pytorch/test/distributed/flight_recorder/ |
| D | test_fr_analysis.py | 23 input_sizes, argument
|
| /external/pytorch/aten/src/ATen/cuda/ |
| D | CUDASparseDescriptors.cpp | 62 IntArrayRef input_sizes = input.sizes(); in createRawDnMatDescriptor() local
145 IntArrayRef input_sizes = input.sizes(); in CuSparseSpMatCsrDescriptor() local
|
| /external/pytorch/aten/src/ATen/mkl/ |
| D | SparseDescriptors.h | 83 IntArrayRef input_sizes = input.sizes(); in MklSparseCsrDescriptor() local
|
| /external/executorch/backends/vulkan/runtime/graph/ops/impl/ |
| D | Upsample.cpp | 67 utils::uvec3 input_sizes = t_in->logical_limits(); in add_upsample_nearest2d_node() local
|
| /external/pytorch/aten/src/ATen/native/nested/cuda/ |
| D | NestedTensorTransformerFunctions.cu | 43 const int* input_sizes, in remove_padding_transform0213_2()
84 const int* input_sizes, in remove_padding_2()
117 const int* input_sizes, in remove_padding()
155 const int* input_sizes, in remove_padding_kernelLauncher()
189 const int* input_sizes, in remove_padding_transform0213_kernelLauncher()
253 const int* input_sizes, in add_padding_1()
291 const int* input_sizes, in add_padding_2()
335 const int* input_sizes, in add_padding_3()
387 const int* input_sizes, in add_padding_kernelLauncher()
|
| /external/pytorch/torch/csrc/autograd/functions/ |
| D | comm.cpp | 96 std::vector<int64_t> input_sizes; in apply() local
|
| /external/pytorch/aten/src/ATen/native/cuda/ |
| D | Activation.cpp | 33 auto input_sizes = input.sizes(); in glu_backward_cuda_out() local
|
| /external/pytorch/aten/src/ATen/functorch/ |
| D | PyTorchOperatorHacks.cpp | 56 const auto input_sizes = input.sizes(); in linear_hack() local
133 auto input_sizes = input.sizes(); in make_feature_noise() local
|
| D | BatchRulesViews.cpp | 322 c10::SymIntArrayRef input_sizes, int64_t offset, int64_t dim1, int64_t dim2) { in diagonal_backward_batch_rule()
394 c10::SymIntArrayRef input_sizes, int64_t dim, c10::SymInt index) { in select_backward_batch_rule()
407 SymIntArrayRef input_sizes, int64_t dim, c10::SymInt start, c10::SymInt end, c10::SymInt step) { in slice_backward_batch_rule()
|
| /external/pytorch/torch/csrc/lazy/ts_backend/ |
| D | ts_native_functions.cpp | 441 c10::SymIntArrayRef input_sizes, in select_backward_symint()
506 at::SymIntArrayRef input_sizes, in diagonal_backward_symint()
516 at::SymIntArrayRef input_sizes, in slice_backward_symint()
|
| /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
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_grad_shape_utils.cc | 174 Status Conv2DBackpropComputeInputShape(const Tensor& input_sizes, in Conv2DBackpropComputeInputShape()
|