Home
last modified time | relevance | path

Searched defs:input_data (Results 26 – 50 of 461) sorted by relevance

12345678910>>...19

/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/sparse_ops/
H A Dfully_connected.h35 const RuntimeShape& input_shape, const float* input_data, in FullyConnectedSparseWeight()
81 const RuntimeShape& input_shape, const int8_t* input_data, in FullyConnectedSparseWeight1x16Impl()
117 const RuntimeShape& input_shape, const float* input_data, in FullyConnectedSparseWeight1x4Impl()
187 const float* input_data; member
201 const RuntimeShape& input_shape, const int8_t* input_data, in FullyConnectedSparseWeight1x16()
225 const RuntimeShape& input_shape, const float* input_data, in FullyConnectedSparseWeight1x4()
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/
H A Dlogistic.h30 inline void Logistic(const RuntimeShape& input_shape, const float* input_data, in Logistic()
62 const float* input_data, const RuntimeShape& output_shape, in Logistic()
69 const RuntimeShape& input_shape, const int16_t* input_data, in Logistic()
93 inline void Logistic(const RuntimeShape& input_shape, const int8_t* input_data, in Logistic()
H A Dtanh.h29 inline void Tanh(const RuntimeShape& input_shape, const float* input_data, in Tanh()
43 const float* input_data, const RuntimeShape& output_shape, in Tanh()
50 const int16_t* input_data, const RuntimeShape& output_shape, in Tanh()
84 const uint8_t* input_data, const RuntimeShape& output_shape, in Tanh()
H A Dsoftmax.h32 const RuntimeShape& input_shape, const float* input_data, in Softmax()
69 const RuntimeShape& input_shape, const InputT* input_data, in Softmax()
150 const int16_t* input_data, const int depth, in SoftMaxCalculateExp()
168 const int16_t* input_data, in SoftmaxInt16()
H A Dpooling.h30 const float* input_data, in AveragePool()
85 const uint8_t* input_data, in AveragePool()
143 const float* input_data, const RuntimeShape& output_shape, in L2Pool()
197 const float* input_data, const RuntimeShape& output_shape, in MaxPool()
247 const uint8_t* input_data, const RuntimeShape& output_shape, in MaxPool()
H A Dlog_softmax.h29 const RuntimeShape& input_shape, const float* input_data, in LogSoftmax()
62 const uint8_t* input_data, in LogSoftmax()
157 const T* input_data, in LogSoftmaxQuantized()
247 const int8_t* input_data, in LogSoftmax()
/aosp_15_r20/external/libjpeg-turbo/simd/arm/aarch64/
H A Djsimd.c441 JSAMPARRAY input_data, JSAMPARRAY output_data) in jsimd_h2v2_downsample()
450 JSAMPARRAY input_data, JSAMPARRAY output_data) in jsimd_h2v1_downsample()
492 JSAMPARRAY input_data, JSAMPARRAY *output_data_ptr) in jsimd_h2v2_upsample()
500 JSAMPARRAY input_data, JSAMPARRAY *output_data_ptr) in jsimd_h2v1_upsample()
559 JSAMPARRAY input_data, JSAMPARRAY *output_data_ptr) in jsimd_h2v2_fancy_upsample()
568 JSAMPARRAY input_data, JSAMPARRAY *output_data_ptr) in jsimd_h2v1_fancy_upsample()
577 JSAMPARRAY input_data, JSAMPARRAY *output_data_ptr) in jsimd_h1v2_fancy_upsample()
/aosp_15_r20/external/libjpeg-turbo/simd/arm/aarch32/
H A Djsimd.c376 JSAMPARRAY input_data, JSAMPARRAY output_data) in jsimd_h2v2_downsample()
385 JSAMPARRAY input_data, JSAMPARRAY output_data) in jsimd_h2v1_downsample()
427 JSAMPARRAY input_data, JSAMPARRAY *output_data_ptr) in jsimd_h2v2_upsample()
435 JSAMPARRAY input_data, JSAMPARRAY *output_data_ptr) in jsimd_h2v1_upsample()
494 JSAMPARRAY input_data, JSAMPARRAY *output_data_ptr) in jsimd_h2v2_fancy_upsample()
503 JSAMPARRAY input_data, JSAMPARRAY *output_data_ptr) in jsimd_h2v1_fancy_upsample()
512 JSAMPARRAY input_data, JSAMPARRAY *output_data_ptr) in jsimd_h1v2_fancy_upsample()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/
H A DUnfold2d.cpp39 scalar_t* input_data, in unfolded2d_acc()
124 scalar_t* input_data, in unfolded2d_acc_channels_last()
181 void *input_data, in unfolded2d_acc_kernel()
231 const scalar_t* input_data, in unfolded2d_copy()
338 const scalar_t* input_data, in unfolded2d_copy_channels_last()
396 const void *input_data, in unfolded2d_copy_kernel()
/aosp_15_r20/external/tensorflow/tensorflow/tools/graph_transforms/
H A Dfold_old_batch_norms_test.cc44 Tensor input_data(DT_FLOAT, TensorShape({1, 1, 6, 2})); in TestFoldOldBatchNorms() local
128 Tensor input_data(DT_FLOAT, TensorShape({1, 1, 6, 2})); in TestFoldOldBatchNormsAfterDepthwiseConv2dNative() local
206 Tensor input_data(DT_FLOAT, TensorShape({1, 1, 6, 2})); in TestFoldFusedBatchNorms() local
283 Tensor input_data(DT_FLOAT, TensorShape({1, 1, 6, 2})); in TestFoldFusedBatchNormsAfterDepthwiseConv2dNative() local
364 Tensor input_data(DT_FLOAT, input_shape); in TestFoldFusedBatchNormsWithConcat() local
462 Tensor input_data(DT_FLOAT, TensorShape({2, 1, 3, 2})); in TestFoldFusedBatchNormsWithBatchToSpace() local
H A Dfold_batch_norms_test.cc43 Tensor input_data(DT_FLOAT, TensorShape({1, 1, 6, 2})); in TestFoldBatchNormsConv2D() local
94 Tensor input_data(DT_FLOAT, TensorShape({1, 1, 6, 2})); in TestFoldBatchNormsDepthwiseConv2dNative() local
145 Tensor input_data(DT_FLOAT, TensorShape({1, 1, 6, 2})); in TestFoldBatchNormsConv2DShared() local
203 Tensor input_data(DT_FLOAT, TensorShape({6, 2})); in TestFoldBatchNormsMatMul() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/
H A Ddnn.h1323 DeviceMemory<ElementType> input_data, in PrepareForConvolution()
1649 const DeviceMemory<float>& input_data, DeviceMemory<float>* output_data) { in DoNormalizeWithDimensions()
1690 const DeviceMemory<float>& input_data, in DoActivate()
1739 port::ArraySlice<const DeviceMemory<float>*> input_data, // non-absl ok in DoSpaceConcatenate()
1775 const DeviceMemory<float>& input_data, in DoReshape()
1803 const DeviceMemory<float>& input_data, in DoDepthToSpace()
1832 const DeviceMemory<float>& input_data, in DoSpaceToDepth()
1888 port::ArraySlice<const DeviceMemory<float>*> input_data, // non-absl ok in DoElementwiseOperateScaledQuantized()
1966 const DeviceMemory<float>& input_data, in DoXYBroadcast()
2115 const DeviceMemory<Eigen::half>& input_data, in DoRnnForward()
[all …]
/aosp_15_r20/external/libjpeg-turbo/simd/arm/
H A Djdsample-neon.c61 JSAMPARRAY input_data, in jsimd_h2v1_fancy_upsample_neon()
206 JSAMPARRAY input_data, in jsimd_h2v2_fancy_upsample_neon()
412 JSAMPARRAY input_data, in jsimd_h1v2_fancy_upsample_neon()
487 JSAMPARRAY input_data, in jsimd_h2v1_upsample_neon()
540 JSAMPARRAY input_data, in jsimd_h2v2_upsample_neon()
/aosp_15_r20/trusty/user/app/sample/hwcryptohal/server/
Dcrypto_operation.rs280 let input_data = input_buffer.read_into_buffer_reference(input, None)?; in update() localVariable
426 let input_data = in cbcs_update() localVariable
501 let input_data = input_buffer.read_into_buffer_reference(input, None)?; in update() localVariable
714 let mut input_data = input_to_encrypt.as_bytes().to_vec(); in use_aes_key() localVariable
803 let mut input_data = input_to_encrypt.as_bytes().to_vec(); in process_aes_encrypt_decrypt_operations() localVariable
818 let mut input_data = input_to_encrypt_2.as_bytes().to_vec(); in process_aes_encrypt_decrypt_operations() localVariable
835 let mut input_data = input_to_encrypt_3.as_bytes().to_vec(); in process_aes_encrypt_decrypt_operations() localVariable
849 let mut input_data = input_to_encrypt_4.as_bytes().to_vec(); in process_aes_encrypt_decrypt_operations() localVariable
863 let mut input_data = input_to_encrypt_5.as_bytes().to_vec(); in process_aes_encrypt_decrypt_operations() localVariable
877 let mut input_data = input_to_encrypt_6.as_bytes().to_vec(); in process_aes_encrypt_decrypt_operations() localVariable
Dcmd_processing.rs1436 let input_data = OperationData::DataBuffer("string to be encrypted".as_bytes().to_vec()); in aes_simple_test() localVariable
1767 let input_data = in aes_simple_cbcs_test() localVariable
1770 let input_data = in aes_simple_cbcs_test() localVariable
1773 let input_data = in aes_simple_cbcs_test() localVariable
1776 let input_data = in aes_simple_cbcs_test() localVariable
1779 let input_data = OperationData::DataBuffer( in aes_simple_cbcs_test() localVariable
1926 let input_data = in aes_simple_all_encrypted_cbcs_test() localVariable
1929 let input_data = in aes_simple_all_encrypted_cbcs_test() localVariable
1958 let input_data = in aes_simple_all_encrypted_cbcs_test() localVariable
1961 let input_data = in aes_simple_all_encrypted_cbcs_test() localVariable
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DLossMultiMargin.cpp23 const scalar_t* input_data, in multi_margin_inner_sum_cpu()
62 const scalar_t* input_data, in multi_margin_loss_cpu_kernel()
134 auto input_data = input_contiguous.const_data_ptr<scalar_t>(); in multi_margin_loss_out_cpu_template() local
155 const scalar_t* input_data, in multi_margin_loss_backward_cpu_kernel()
239 auto input_data = input_contiguous.const_data_ptr<scalar_t>(); in multi_margin_loss_backward_out_cpu_template() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Deigen_benchmark.h52 Scalar* input_data = in SpatialConvolution() local
129 Scalar* input_data = in SpatialConvolutionBackwardKernel() local
184 Scalar* input_data = in CuboidConvolution() local
265 Scalar* input_data = in CuboidConvolutionBackwardKernel() local
/aosp_15_r20/external/libjpeg-turbo/simd/x86_64/
H A Djsimd.c343 JSAMPARRAY input_data, JSAMPARRAY output_data) in jsimd_h2v2_downsample()
362 JSAMPARRAY input_data, JSAMPARRAY output_data) in jsimd_h2v1_downsample()
419 JSAMPARRAY input_data, JSAMPARRAY *output_data_ptr) in jsimd_h2v2_upsample()
434 JSAMPARRAY input_data, JSAMPARRAY *output_data_ptr) in jsimd_h2v1_upsample()
491 JSAMPARRAY input_data, JSAMPARRAY *output_data_ptr) in jsimd_h2v2_fancy_upsample()
508 JSAMPARRAY input_data, JSAMPARRAY *output_data_ptr) in jsimd_h2v1_fancy_upsample()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/
H A Dreduce_test.cc104 std::vector<float> input_data(element_count); in RunR1ToR0Test() local
121 void RunR1ToR0PredTest(bool and_reduce, absl::Span<const int> input_data) { in RunR1ToR0PredTest()
179 Array2D<uint8_t> input_data(rows, cols); in RunR2ToR1PredTest() local
213 Array2D<float> input_data(rows, cols); in RunR2ToR0Test() local
241 Array2D<float> input_data(rows, cols); in RunR2ToR1Test() local
299 Array2D<NativeT> input_data(rows, cols); in RunVectorizedReduceTestForType() local
453 Array2D<float> input_data(rows, cols); in XLA_TEST_F() local
485 Array2D<float> input_data(rows, cols); in XLA_TEST_F() local
533 Array3D<float> input_data(rows, 2, cols / 2); in XLA_TEST_F() local
829 std::unique_ptr<GlobalData> input_data = in XLA_TEST_P() local
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/
H A Dmkl_fused_ops_test.cc217 void RunConv2DUnfused(const Tensor& input_data, const Tensor& filter_data, in RunConv2DUnfused()
338 [this](const Tensor& input_data, const Tensor& filter_data, in VerifyFusedConv2D()
340 Tensor* out, const int padding) { in VerifyFusedConv2D()
534 void RunDepthwiseConv2DUnfused(const Tensor& input_data, in RunDepthwiseConv2DUnfused()
629 [this](const Tensor& input_data, const Tensor& filter_data, in VerifyFusedDepthwiseConv2D()
631 Tensor* out, const int padding) { in VerifyFusedDepthwiseConv2D()
1294 [this](const Tensor& input_data, const Tensor& filter_data, in VerifyPadAndConv2DWithBias()
1312 [this](const Tensor& input_data, const Tensor& filter_data, in VerifyPadAndConv2DWithBiasRelu()
1319 [this](const Tensor& input_data, const Tensor& filter_data, in VerifyPadAndConv2DWithBiasRelu()
1330 void RunMklPadWithFusedConv2DAndBias(const Tensor& input_data, in RunMklPadWithFusedConv2DAndBias()
[all …]
/aosp_15_r20/external/libjpeg-turbo/simd/i386/
H A Djsimd.c389 JSAMPARRAY input_data, JSAMPARRAY output_data) in jsimd_h2v2_downsample()
412 JSAMPARRAY input_data, JSAMPARRAY output_data) in jsimd_h2v1_downsample()
477 JSAMPARRAY input_data, JSAMPARRAY *output_data_ptr) in jsimd_h2v2_upsample()
495 JSAMPARRAY input_data, JSAMPARRAY *output_data_ptr) in jsimd_h2v1_upsample()
559 JSAMPARRAY input_data, JSAMPARRAY *output_data_ptr) in jsimd_h2v2_fancy_upsample()
580 JSAMPARRAY input_data, JSAMPARRAY *output_data_ptr) in jsimd_h2v1_fancy_upsample()
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/
H A Dsoftmax_quantized_test.cc35 void RunSoftmaxFloatReference(const uint8* input_data, in RunSoftmaxFloatReference()
106 void RunOneSoftmaxTest(const uint8* input_data, in RunOneSoftmaxTest()
190 std::vector<uint8> input_data(buffer_size); in TryOneUniformSoftmax() local
228 std::vector<uint8> input_data(buffer_size); in TryOneSkyscraperSoftmax() local
/aosp_15_r20/external/tensorflow/tensorflow/stream_executor/rocm/
H A Drocm_dnn.cc1106 const void* input_data, in Execute()
2029 const DeviceMemory<T>& input_data, in ExtractAndCheckRnnForward()
2139 const DeviceMemory<T>& input_data, in DoRnnForwardImpl()
2254 const DeviceMemory<T>& input_data, in DoRnnBackwardImpl()
2634 const DeviceMemory<Eigen::half>& input_data, in DoRnnForward()
2678 const DeviceMemory<float>& input_data, in DoRnnForward()
2721 const DeviceMemory<double>& input_data, in DoRnnForward()
2744 const DeviceMemory<Eigen::half>& input_data, in DoRnnBackward()
2797 const DeviceMemory<float>& input_data, in DoRnnBackward()
2849 const DeviceMemory<double>& input_data, in DoRnnBackward()
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
H A Dresize_bilinear.h267 const uint8* input_data, in ResizeBilinear888Uint8()
1329 const float* input_data, in ResizeBilinearKernel2x2()
1496 const float* input_data, in ResizeBilinear2x2()
1514 float width_scale, const RuntimeShape& input_shape, const float* input_data, in ResizeBilinearGeneric()
1567 float width_scale, const RuntimeShape& input_shape, const T* input_data, in ResizeBilinearGenericSmallChannel()
1611 const float* input_data, in ResizeBilinear()
1662 const uint8* input_data, in ResizeBilinear()
1721 const int8* input_data, in ResizeBilinear()
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/
H A Dfully_connected.h35 const int8_t* input_data, const RuntimeShape& filter_shape, in FullyConnectedPerChannel()
77 const int16_t* input_data, const RuntimeShape& filter_shape, in FullyConnectedPerChannel()
115 const int8_t* input_data, const RuntimeShape& filter_shape, in FullyConnected()
159 const int16_t* input_data, const RuntimeShape& filter_shape, in FullyConnected()

12345678910>>...19