Home
last modified time | relevance | path

Searched defs:input_batch (Results 1 – 25 of 33) sorted by relevance

12

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/
H A Druntime_fft_impl.h40 int64_t input_batch, int64_t fft_length0, int64_t fft_length1, in EigenFftC2C()
67 int64_t input_batch, int64_t fft_length0, int64_t fft_length1, in EigenFftR2C()
106 int64_t input_batch, int64_t fft_length0, int64_t fft_length1, in EigenFftC2R()
178 int64_t input_batch, int64_t fft_length0, in EigenFftWithRank()
242 int32_t fft_rank, int64_t input_batch, int64_t fft_length0, in EigenFftImpl()
H A Druntime_conv2d_acl.cc31 int64_t input_batch, int64_t input_rows, int64_t input_cols, in ACLDepthwiseConvImpl()
128 float* rhs, int64_t input_batch, int64_t input_rows, in ACLGemmConvImpl()
235 int64_t input_batch, int64_t input_rows, int64_t input_cols, in __xla_cpu_runtime_ACLConv2DF32()
H A Druntime_single_threaded_conv2d.cc24 Eigen::half* rhs, int64_t input_batch, int64_t input_rows, in __xla_cpu_runtime_EigenSingleThreadedConv2DF16()
44 int64_t input_batch, int64_t input_rows, int64_t input_cols, in __xla_cpu_runtime_EigenSingleThreadedConv2DF32()
H A Druntime_conv2d.cc27 int64_t input_batch, int64_t input_rows, int64_t input_cols, in __xla_cpu_runtime_EigenConv2DF32()
49 Eigen::half* rhs, int64_t input_batch, int64_t input_rows, in __xla_cpu_runtime_EigenConv2DF16()
H A Druntime_conv3d.cc27 int64_t input_batch, int64_t input_x, int64_t input_y, int64_t input_z, in __xla_cpu_runtime_EigenConv3DF32()
52 Eigen::half* rhs, int64_t input_batch, int64_t input_x, int64_t input_y, in __xla_cpu_runtime_EigenConv3DF16()
H A Druntime_single_threaded_conv3d.cc24 int64_t input_batch, int64_t input_x, int64_t input_y, int64_t input_z, in __xla_cpu_runtime_EigenSingleThreadedConv3DF32()
47 Eigen::half* rhs, int64_t input_batch, int64_t input_x, int64_t input_y, in __xla_cpu_runtime_EigenSingleThreadedConv3DF16()
H A Druntime_conv2d_mkl.cc53 ScalarType* rhs, int64_t input_batch, int64_t input_rows, in MKLConvImpl()
156 int64_t input_batch, int64_t input_rows, int64_t input_cols, in __xla_cpu_runtime_MKLConv2DF32()
H A Druntime_conv_impl.h32 ScalarType* rhs, Eigen::Index input_batch, Eigen::Index input_x, in EigenConv2DImpl()
109 ScalarType* rhs, Eigen::Index input_batch, Eigen::Index input_x, in EigenConv3DImpl()
H A Druntime_single_threaded_fft.cc23 int32_t double_precision, int32_t fft_rank, int64_t input_batch, in __xla_cpu_runtime_EigenSingleThreadedFft()
H A Druntime_fft.cc27 int32_t double_precision, int32_t fft_rank, int64_t input_batch, in __xla_cpu_runtime_EigenFft()
H A Druntime_conv2d_acl.h79 float* lhs, float* rhs, int64_t input_batch, int64_t input_rows, in __xla_cpu_runtime_ACLConv2DF32()
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Ddepthwise_conv_test.cc475 const int input_batch = 2; in BatchPaddingValidTest() local
536 const int input_batch = 4; in BatchPaddingSameTest() local
1069 const int input_batch = 1; in TEST_P() local
1142 const int input_batch = 1; in TEST_P() local
1217 const int input_batch = 2; in TEST_P() local
1281 const int input_batch = 2; in TEST_P() local
1354 const int input_batch = 1; in TEST_P() local
1418 const int input_batch = 1; in TEST_P() local
1488 const int input_batch = 1; in TEST_P() local
1546 const int input_batch = 1; in TEST_P() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dquantized_batch_norm_op_test.cc62 const int input_batch = 1; in TEST_F() local
159 const int input_batch = 1; in TEST_F() local
H A Dlist_kernels.h1102 auto input_batch = input_reshaped.Slice(b, b + 1); in Compute() local
/aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/data/transforms/
H A Dbase.h42 std::vector<Output> apply_batch(std::vector<Input> input_batch) override { in apply_batch()
H A Dlambda.h26 OutputBatchType apply_batch(InputBatchType input_batch) override { in apply_batch()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DFractionalMaxPooling.h48 int64_t input_batch = 0, input_channel = 0; in fractional_max_pool_check_shape() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/
H A Dspace_to_depth.h41 const int input_batch = input_shape.Dims(0); in SpaceToDepth() local
H A Ddepth_to_space.h39 const int input_batch = input_shape.Dims(0); in DepthToSpace() local
H A Dspace_to_batch_nd.h80 int input_batch = out_b % input_batch_size; in SpaceToBatchND() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/
H A Dconvolution_dimension_numbers_test.cc35 int64_t input_batch, int64_t input_feature, int64_t input_first_spatial, in CreateConvDimensionNumbers()
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/
H A Dmkl_conv_ops.h130 int input_batch = static_cast<int>(input_batch_raw); in GetInputSizeInMklOrder() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dconvolution_group_converter.cc233 const int64_t input_batch = in HandleBatchGroupCount() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/batching_util/
H A Dshared_batch_scheduler.h830 auto input_batch = std::make_shared<BatchInputTask<TaskType>>( in ScheduleWithLazySplit() local
/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/arm_conv/depthwise/
H A Ddepthwise_planar.hpp366 auto input_batch = reinterpret_cast<const TInput *>(input); in execute_internal() local

12