Home
last modified time | relevance | path

Searched defs:input_t (Results 1 – 25 of 43) sorted by relevance

12

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dencode_proto_op.cc78 auto input_t = input.flat_inner_dims<int64_t>(); in TotalPackedSize() local
91 auto input_t = input.flat_inner_dims<uint64>(); in TotalPackedSize() local
104 auto input_t = input.flat_inner_dims<int64_t>(); in TotalPackedSize() local
117 auto input_t = input.flat_inner_dims<int32>(); in TotalPackedSize() local
155 auto input_t = input.flat_inner_dims<uint64>(); in TotalPackedSize() local
168 auto input_t = input.flat_inner_dims<uint32>(); in TotalPackedSize() local
181 auto input_t = input.flat_inner_dims<int32>(); in TotalPackedSize() local
212 auto input_t = input.flat_inner_dims<int32>(); in TotalPackedSize() local
224 auto input_t = input.flat_inner_dims<int64_t>(); in TotalPackedSize() local
236 auto input_t = input.flat_inner_dims<int64_t>(); in TotalPackedSize() local
[all …]
H A Dpopulation_count_op.cc46 const Tensor& input_t = c->input(0); in Compute() local
/aosp_15_r20/external/pytorch/test/distributed/
H A Dtest_c10d_ops_nccl.py418 def allgather_base(output_t, input_t): argument
440 def allgather_base(output_t, input_t): argument
474 def gather(output_t, input_t, rootRank): argument
509 def gather(output_t, input_t, rootRank): argument
587 def scatter(output_t, input_t, rootRank): argument
622 def scatter(output_t, input_t, rootRank): argument
705 def reduce_scatter_base(output_t, input_t): argument
857 def reduce_scatter_base(output_t, input_t): argument
961 def allgather_base(output_t, input_t): argument
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DSummaryOps.cu37 at::acc_type<input_t, /*is_cuda=*/true> minvalue, in getBin()
178 at::acc_type<input_t, /*is_cuda=*/true> minvalue, in CUDA_tensor_histogram()
312 at::acc_type<input_t, /*is_cuda=*/true> min, in _histc_cuda_template()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/mkldnn/
H A DConv.cpp166 const Tensor& input_t, in _mkldnn_convolution_out()
214 const Tensor& input_t, in _mkldnn_convolution()
279 const Tensor& input_t, in mkldnn_convolution()
300 const Tensor& input_t, in mkldnn_convolution_pointwise()
336 const Tensor& input_t, in mkldnn_convolution_pointwise_binary()
501 const Tensor& input_t, in mkldnn_convolution_pointwise_binary_()
619 const Tensor& input_t, in _mkldnn_convolution_transpose()
715 const Tensor& input_t, in mkldnn_convolution_transpose_pointwise()
746 const Tensor& input_t, in mkldnn_convolution_transpose_pointwise_meta()
866 const Tensor& input_t, const Tensor& grad_output_t, const Tensor& weight_t, in mkldnn_convolution_backward()
[all …]
H A DLinear.cpp184 const Tensor& input_t, in mkldnn_linear_pointwise()
258 const Tensor& input_t, in mkldnn_linear_pointwise_binary()
H A DUtils.h137 inline void mkldnn_check_low_precision(ScalarType input_t, std::string name) { in mkldnn_check_low_precision()
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Dremapper_test.cc161 auto input_t = GenerateRandomTensor<DT_FLOAT>({2, 8, 8, num_channels}); in TEST_F() local
269 auto input_t = GenerateRandomTensor<DT_FLOAT>({2, 8, 8, num_channels}); in TEST_F() local
398 auto input_t = GenerateRandomTensor<DT_FLOAT>(input_shape); in TEST_F() local
535 auto input_t = GenerateRandomTensor<DT_FLOAT>(input_shape); in TEST_F() local
642 auto input_t = GenerateTensorWithSetRandom<DTYPE>({8, 32, 32, 3}); in RunTest() local
749 auto input_t = GenerateTensorWithSetRandom<DTYPE>({8, 32, 32, 3}); in RunTest() local
903 auto input_t = GenerateTensorWithSetRandom<DTYPE>({8, 32, 1, 3}); in RunTest() local
1036 auto input_t = GenerateRandomTensor<DT_FLOAT>({8, 4, 32, 32, 3}); in TEST_F() local
1102 auto input_t = GenerateRandomTensor<DT_FLOAT>({8, 4, 32, 32, 3}); in TEST_F() local
1168 auto input_t = GenerateRandomTensor<DT_FLOAT>({8, 32, 32, 3}); in TEST_F() local
[all …]
H A Dmkl_remapper_test.cc353 auto input_t = GenerateRandomTensor<DT_FLOAT>({2, 8, 8, num_channels}); in TEST_F() local
581 auto input_t = GenerateRandomTensor<DTYPE>({8, 32, 32, 3}); in RelpaceAddWithBiasAddDepthwiseConv2D() local
782 Tensor input_t = Tensor(DataTypeToEnum<T>::v(), input_shape); in VerifyFused() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cudnn/
H A DConvShared.cpp241 const Tensor& input_t, in cudnn_convolution()
277 const Tensor& input_t, in cudnn_convolution_out()
469 const Tensor& input_t, in cudnn_convolution_transpose()
506 const Tensor& input_t, in cudnn_convolution_backward_weight()
559 const Tensor& input_t, in cudnn_convolution_backward_weight()
638 const Tensor& input_t, in cudnn_convolution_transpose_backward_weight()
706 const Tensor& input_t, in cudnn_convolution_relu()
758 const Tensor& input_t, in cudnn_convolution_add_relu()
H A DGridSampler.cpp23 Tensor cudnn_grid_sampler_forward(const Tensor& input_t, const Tensor& grid_t) { in cudnn_grid_sampler_forward()
28 const Tensor& input_t, in cudnn_grid_sampler_backward()
80 Tensor cudnn_grid_sampler_forward(const Tensor& input_t, const Tensor& grid_t) { in cudnn_grid_sampler_forward()
128 const Tensor& input_t, in cudnn_grid_sampler_backward()
H A DBatchNorm.cpp45 const Tensor& input_t, in _get_cudnn_batch_norm_reserve_space_size()
107 const Tensor& input_t, in _get_cudnn_batch_norm_reserve_space_size()
122 const Tensor& input_t, in cudnn_batch_norm()
272 const Tensor& input_t, in cudnn_batch_norm_backward()
H A DConvPlaceholders.cpp42 const Tensor& input_t, in cudnn_convolution_out()
206 const Tensor& input_t, in cudnn_convolution_relu()
217 const Tensor& input_t, in cudnn_convolution_add_relu()
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/
H A Dmkl_conv_ops_test.cc105 Tensor input_t = GetRandomInputTensor(dims); in DefaultConv2D() local
126 Tensor input_t = GetRandomInputTensor(dims); in MklConv2D() local
214 Tensor input_t = GetRandomInputTensor(dims); in DefaultConv2DBwdFilter() local
242 Tensor input_t = GetRandomInputTensor(dims); in MklConv2DBwdFilter() local
H A Dmkl_swish_op_test.cc60 Tensor input_t(dtype, shape); in SwishGraph() local
H A Dmkl_relu_op_test.cc53 Tensor input_t(DT_FLOAT, shape); in Activation() local
/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/array_ops/
H A Dslice_op_test.py188 def func(input_t): argument
204 def func(input_t): argument
/aosp_15_r20/external/pytorch/aten/src/ATen/native/miopen/
H A DBatchNorm_miopen.cpp61 …const Tensor& input_t, const Tensor& weight_t, const std::optional<Tensor>& bias_t_opt, const std:… in miopen_batch_norm()
160 const Tensor& input_t, in miopen_batch_norm_backward()
H A DConv_miopen.cpp798 const Tensor& input_t, const Tensor& weight_t, const std::optional<Tensor>& bias_t_opt, in miopen_convolution()
899 const Tensor& input_t, const Tensor& weight_t, const std::optional<Tensor>& bias_t_opt, in miopen_depthwise_convolution()
1102 const Tensor& input_t, in miopen_depthwise_convolution_backward_weight()
1155 const Tensor& input_t, in miopen_convolution_backward_weight()
1182 const Tensor& input_t, in miopen_convolution_transpose_backward_weight()
1466 const Tensor& input_t, const Tensor& weight_t, const std::optional<Tensor>& bias_t_opt, in miopen_convolution_transpose()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/
H A DCopyKernel.cpp18 static bool reduced_input(ScalarType input_t, ScalarType output_t) { in reduced_input()
23 static bool reduced_output(ScalarType input_t, ScalarType output_t) { in reduced_output()
/aosp_15_r20/external/tensorflow/tensorflow/c/eager/
H A Dc_api_unified_experimental_test.cc413 TF_AbstractTensor* input_t = in TEST_P() local
510 TF_AbstractTensor* input_t = in TEST_P() local
/aosp_15_r20/external/pytorch/torch/csrc/autograd/
H A DFunctionsManual.cpp1291 const Tensor& input_t, in convolution_jvp()
1329 const Tensor& input_t, in _convolution_jvp()
4790 const Tensor& input_t, in layer_norm_double_backward()
6104 const Tensor& input_t, in _invstd_jvp()
6129 const Tensor& input_t, in _norm_jvp()
6156 Tensor& input_t, in _affine_jvp()
6191 const Tensor& input_t, in batch_norm_jvp()
6246 const Tensor& input_t, in layer_norm_jvp()
6285 const Tensor& input_t, in group_norm_jvp()
6335 const Tensor& input_t, in group_norm_mean_jvp()
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/shim/test_op/
H A Dsimple_op.h144 const auto& input_t = input_t_or.value(); in ShapeInference() local
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Dshape_inference_test.cc202 Tensor input_t = in TEST_F() local
228 Tensor input_t = ::tensorflow::test::AsTensor<int32>({1, 2, 3, 4, 5}); in TEST_F() local
254 Tensor input_t = ::tensorflow::test::AsTensor<int32>({1, 2, 3, 4, 5}); in TEST_F() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DConvolutionMM2d.cpp69 auto input_t = input_a[t]; in compute_columns2d() local
603 auto input_t = input_a[t]; in slow_conv2d_forward_out_cpu() local

12