Home
last modified time | relevance | path

Searched refs:input_batch (Results 1 – 25 of 46) sorted by relevance

12

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Druntime_fft_impl.h40 int64_t input_batch, int64_t fft_length0, int64_t fft_length1, in EigenFftC2C() argument
50 dims[0] = input_batch; in EigenFftC2C()
67 int64_t input_batch, int64_t fft_length0, int64_t fft_length1, in EigenFftR2C() argument
73 in_dims[0] = input_batch; in EigenFftR2C()
75 out_dims[0] = input_batch; in EigenFftR2C()
106 int64_t input_batch, int64_t fft_length0, int64_t fft_length1, in EigenFftC2R() argument
112 in_dims[0] = input_batch; in EigenFftC2R()
114 out_dims[0] = input_batch; in EigenFftC2R()
178 int64_t input_batch, int64_t fft_length0, in EigenFftWithRank() argument
185 static_cast<complex128*>(operand), input_batch, fft_length0, in EigenFftWithRank()
[all …]
Druntime_conv_impl.h32 ScalarType* rhs, Eigen::Index input_batch, Eigen::Index input_x, in EigenConv2DImpl() argument
43 input(lhs, input_batch, input_x, input_y, input_channels); in EigenConv2DImpl()
51 output(out, input_batch, output_x, output_y, kernel_filters); in EigenConv2DImpl()
57 input_reshaped_dims[0] = input_batch; in EigenConv2DImpl()
64 output_reshaped_dims[0] = input_batch; in EigenConv2DImpl()
75 pre_contract_dims[0] = output_y * output_x * input_batch; in EigenConv2DImpl()
80 post_contract_dims[0] = input_batch; in EigenConv2DImpl()
109 ScalarType* rhs, Eigen::Index input_batch, Eigen::Index input_x, in EigenConv3DImpl() argument
125 const ConstTType input(lhs, input_batch, input_x, input_y, input_z, in EigenConv3DImpl()
133 output(out, input_batch, output_x, output_y, output_z, kernel_filters); in EigenConv3DImpl()
[all …]
Druntime_conv2d_acl.cc31 int64_t input_batch, int64_t input_rows, int64_t input_cols, in ACLDepthwiseConvImpl() argument
65 input_batch), in ACLDepthwiseConvImpl()
72 input_batch), in ACLDepthwiseConvImpl()
128 float* rhs, int64_t input_batch, int64_t input_rows, in ACLGemmConvImpl() argument
166 input_batch), in ACLGemmConvImpl()
174 input_batch), in ACLGemmConvImpl()
235 int64_t input_batch, int64_t input_rows, int64_t input_cols, in __xla_cpu_runtime_ACLConv2DF32() argument
251 run_options_ptr, out, lhs, rhs, input_batch, input_rows, input_cols, in __xla_cpu_runtime_ACLConv2DF32()
259 if (ACLGemmConvImpl(run_options_ptr, out, lhs, rhs, input_batch, input_rows, in __xla_cpu_runtime_ACLConv2DF32()
272 run_options_ptr, out, lhs, rhs, input_batch, input_rows, input_cols, in __xla_cpu_runtime_ACLConv2DF32()
Druntime_single_threaded_conv2d.cc24 Eigen::half* rhs, int64_t input_batch, int64_t input_rows, in __xla_cpu_runtime_EigenSingleThreadedConv2DF16() argument
33 Eigen::DefaultDevice(), out, lhs, rhs, input_batch, 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() argument
53 Eigen::DefaultDevice(), out, lhs, rhs, input_batch, input_rows, in __xla_cpu_runtime_EigenSingleThreadedConv2DF32()
Druntime_conv2d.cc27 int64_t input_batch, int64_t input_rows, int64_t input_cols, in __xla_cpu_runtime_EigenConv2DF32() argument
39 *run_options->intra_op_thread_pool(), out, lhs, rhs, input_batch, in __xla_cpu_runtime_EigenConv2DF32()
49 Eigen::half* rhs, int64_t input_batch, int64_t input_rows, in __xla_cpu_runtime_EigenConv2DF16() argument
61 *run_options->intra_op_thread_pool(), out, lhs, rhs, input_batch, in __xla_cpu_runtime_EigenConv2DF16()
Druntime_conv2d_mkl.cc53 ScalarType* rhs, int64_t input_batch, int64_t input_rows, in MKLConvImpl() argument
70 memory::dims conv1_src_dim = {ToInt(input_batch), ToInt(input_channels), in MKLConvImpl()
75 memory::dims conv1_dst_dim = {ToInt(input_batch), ToInt(kernel_filters), in MKLConvImpl()
156 int64_t input_batch, int64_t input_rows, int64_t input_cols, in __xla_cpu_runtime_MKLConv2DF32() argument
168 run_options_ptr, out, lhs, rhs, input_batch, input_rows, input_cols, in __xla_cpu_runtime_MKLConv2DF32()
174 MKLConvImpl(nullptr, out, lhs, rhs, input_batch, input_rows, input_cols, in __xla_cpu_runtime_MKLConv2DF32()
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() argument
35 Eigen::DefaultDevice(), out, lhs, rhs, input_batch, input_x, input_y, 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() argument
58 Eigen::DefaultDevice(), out, lhs, rhs, input_batch, input_x, input_y, in __xla_cpu_runtime_EigenSingleThreadedConv3DF16()
Druntime_conv3d.cc27 int64_t input_batch, int64_t input_x, int64_t input_y, int64_t input_z, in __xla_cpu_runtime_EigenConv3DF32() argument
41 *run_options->intra_op_thread_pool(), out, lhs, rhs, input_batch, input_x, 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() argument
66 *run_options->intra_op_thread_pool(), out, lhs, rhs, input_batch, input_x, in __xla_cpu_runtime_EigenConv3DF16()
Druntime_single_threaded_fft.cc23 int32_t double_precision, int32_t fft_rank, int64_t input_batch, in __xla_cpu_runtime_EigenSingleThreadedFft() argument
27 static_cast<bool>(double_precision), fft_rank, input_batch, in __xla_cpu_runtime_EigenSingleThreadedFft()
Druntime_fft.cc27 int32_t double_precision, int32_t fft_rank, int64_t input_batch, in __xla_cpu_runtime_EigenFft() argument
34 static_cast<bool>(double_precision), fft_rank, input_batch, in __xla_cpu_runtime_EigenFft()
Druntime_conv2d.h27 float* lhs, float* rhs, int64_t input_batch, int64_t input_rows,
38 Eigen::half* out, Eigen::half* lhs, Eigen::half* rhs, int64_t input_batch,
Druntime_single_threaded_conv2d.h27 Eigen::half* out, Eigen::half* lhs, Eigen::half* rhs, int64_t input_batch,
39 float* lhs, float* rhs, int64_t input_batch, int64_t input_rows,
Druntime_conv3d.h27 Eigen::half* out, Eigen::half* lhs, Eigen::half* rhs, int64_t input_batch,
40 float* lhs, float* rhs, int64_t input_batch, int64_t input_x,
Druntime_single_threaded_conv3d.h27 Eigen::half* out, Eigen::half* lhs, Eigen::half* rhs, int64_t input_batch,
40 float* lhs, float* rhs, int64_t input_batch, int64_t input_x,
Druntime_conv2d_acl.h66 float* lhs, float* rhs, int64_t input_batch, int64_t input_rows,
79 float* lhs, float* rhs, int64_t input_batch, int64_t input_rows, in __xla_cpu_runtime_ACLConv2DF32() argument
/external/pytorch/aten/src/ATen/native/
DFractionalMaxPooling.h48 int64_t input_batch = 0, input_channel = 0; in fractional_max_pool_check_shape() local
52 input_batch = 1; in fractional_max_pool_check_shape()
55 input_batch = input.size(0); in fractional_max_pool_check_shape()
61 input_batch = 1; in fractional_max_pool_check_shape()
64 input_batch = input.size(0); in fractional_max_pool_check_shape()
70 N >= input_batch, in fractional_max_pool_check_shape()
/external/pytorch/torch/csrc/api/include/torch/data/transforms/
Dbase.h22 virtual OutputBatch apply_batch(InputBatch input_batch) = 0;
42 std::vector<Output> apply_batch(std::vector<Input> input_batch) override { in apply_batch() argument
44 output_batch.reserve(input_batch.size()); in apply_batch()
45 for (auto&& input : input_batch) { in apply_batch()
Dlambda.h26 OutputBatchType apply_batch(InputBatchType input_batch) override { in apply_batch() argument
27 return function_(std::move(input_batch)); in apply_batch()
/external/tensorflow/tensorflow/lite/kernels/
Ddepthwise_conv_test.cc475 const int input_batch = 2; in BatchPaddingValidTest() local
485 {input_batch, input_height, input_width, input_depth}}, in BatchPaddingValidTest()
536 const int input_batch = 4; in BatchPaddingSameTest() local
546 {input_batch, input_height, input_width, input_depth}}, in BatchPaddingSameTest()
1069 const int input_batch = 1; in TEST_P() local
1080 {input_batch, input_height, input_width, input_depth}, in TEST_P()
1142 const int input_batch = 1; in TEST_P() local
1153 {input_batch, input_height, input_width, input_depth}, in TEST_P()
1217 const int input_batch = 2; in TEST_P() local
1228 {input_batch, input_height, input_width, input_depth}, in TEST_P()
[all …]
/external/pytorch/test/distributed/_tools/
Dtest_mem_tracker.py59 input_batch = torch.randn(bsz, dim, device=dev, dtype=dtype)
61 mem_tracker.track_external(model, optim, input_batch)
64 model(input_batch).sum().backward()
132 input_batch = torch.randn(bsz, dim, dim, device=dev, dtype=dtype)
134 model(input_batch).sum().backward()
219 input_batch = torch.randn(bsz, dim, device=dev, dtype=dtype)
222 output = model(input_batch)
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dspace_to_depth.h41 const int input_batch = input_shape.Dims(0); in SpaceToDepth() local
53 TFLITE_DCHECK_EQ(input_batch, output_batch); in SpaceToDepth()
55 for (int in_b = 0; in_b < input_batch; ++in_b) { in SpaceToDepth()
Ddepth_to_space.h39 const int input_batch = input_shape.Dims(0); in DepthToSpace() local
51 TFLITE_DCHECK_EQ(input_batch, output_batch); in DepthToSpace()
Dspace_to_batch_nd.h80 int input_batch = out_b % input_batch_size; in SpaceToBatchND() local
96 Offset(input1_shape, input_batch, in SpaceToBatchND()
/external/tensorflow/tensorflow/core/kernels/
Dquantized_batch_norm_op_test.cc62 const int input_batch = 1; in TEST_F() local
67 {input_batch, input_height, input_width, input_depth}); in TEST_F()
121 TensorShape({input_batch, input_height, input_width, input_depth})); in TEST_F()
159 const int input_batch = 1; in TEST_F() local
164 {input_batch, input_height, input_width, input_depth}); in TEST_F()
218 TensorShape({input_batch, input_height, input_width, input_depth})); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dconvolution_dimension_numbers_test.cc35 int64_t input_batch, int64_t input_feature, int64_t input_first_spatial, in CreateConvDimensionNumbers() argument
41 dimension_numbers.set_input_batch_dimension(input_batch); in CreateConvDimensionNumbers()

12