| /aosp_15_r20/external/tensorflow/tensorflow/python/distribute/ |
| H A D | metrics_v1_test.py | 142 def _expected_fn(num_batches): argument
155 def _expected_fn(num_batches): argument
171 def _expected_fn(num_batches): argument
190 def _expected_fn(num_batches): argument
208 def _expected_fn(num_batches): argument
227 def _expected_fn(num_batches): argument
241 def _expected_fn(num_batches): argument
254 def _expected_fn(num_batches): argument
267 def _expected_fn(num_batches): argument
280 def _expected_fn(num_batches): argument
[all …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | eigen_backward_spatial_convolutions_test.cc | 231 const int num_batches = 13; in TEST() local
286 const int num_batches = 13; in TEST() local
341 const int num_batches, const int input_depth, const int input_rows, in test_batched_strided_spatial_convolution_backward_input_valid()
398 int num_batches = 1; in TEST() local
417 const int num_batches, const int input_depth, const int input_rows, in test_batched_strided_spatial_convolution_backward_input_valid_row_major()
475 int num_batches = 1; in TEST() local
494 const int num_batches = 5; in TEST() local
548 const int num_batches = 5; in TEST() local
604 const int num_batches = 11; in TEST() local
651 const int num_batches = 11; in TEST() local
[all …]
|
| H A D | eigen_pooling_test.cc | 31 const int num_batches = 13; in TEST() local
79 const int num_batches = 13; in TEST() local
129 const int num_batches = 13; in TEST() local
189 const int num_batches = 13; in TEST() local
249 const int num_batches = 13; in TEST() local
311 const int num_batches = 13; in TEST() local
373 const int num_batches = 13; in TEST() local
451 const int num_batches = 13; in TEST() local
528 const int num_batches = 13; in TEST() local
576 const int num_batches = 13; in TEST() local
[all …]
|
| H A D | random_binomial_op_test.cc | 27 static Graph* RandomBinomialGraph(double count, double prob, int num_batches, in RandomBinomialGraph()
48 static Graph* RandomBinomialInv(int num_batches, int samples_per_batch) { in RandomBinomialInv()
53 static Graph* RandomBinomialRej(int num_batches, int samples_per_batch) { in RandomBinomialRej()
58 static Graph* RandomBinomialInvComplement(int num_batches, in RandomBinomialInvComplement()
64 static Graph* RandomBinomialRejComplement(int num_batches, in RandomBinomialRejComplement()
|
| H A D | parameterized_truncated_normal_op_test.cc | 27 static Graph* PTruncatedNormal(int num_batches, int samples_per_batch) { in PTruncatedNormal()
56 static Graph* PTruncatedNormal2SD(int num_batches, int samples_per_batch) { in PTruncatedNormal2SD()
83 static Graph* PTruncatedNormalOneTail(int num_batches, int samples_per_batch) { in PTruncatedNormalOneTail()
|
| H A D | eigen_backward_cuboid_convolutions_test.cc | 299 const int num_batches = 13; in TEST() local
366 const int num_batches = 13; in TEST() local
565 const int num_batches = 13; in TEST() local
632 const int num_batches = 13; in TEST() local
699 const int num_batches = 13; in TEST() local
778 const int num_batches = 13; in TEST() local
857 const int num_batches = 13; in TEST() local
934 const int num_batches = 13; in TEST() local
|
| H A D | count_ops.cc | 44 int num_batches = per_batch_counts.size(); in OutputSparse() local
265 int num_batches = is_1d ? 1 : shape_vector(0); in Compute() local
343 int num_batches = splits.NumElements() - 1; in Compute() local
|
| H A D | eigen_spatial_convolutions_test.cc | 197 const int num_batches = 13; in TEST() local
249 const int num_batches = 13; in TEST() local
307 const int num_batches = 13; in TEST() local
363 const int num_batches = 13; in TEST() local
411 const int num_batches = 5; in TEST() local
459 const int num_batches = 13; in TEST() local
510 const int num_batches = 13; in TEST() local
561 const int num_batches = 13; in TEST() local
615 const int num_batches = 13; in TEST() local
|
| H A D | random_binomial_op.cc | 387 int64_t num_batches = 1; in Compute() local
499 int64_t num_batches = 1; in Compute() local
|
| H A D | parameterized_truncated_normal_op.cc | 637 int32_t num_batches = tensor_shape.dim_size(0); in Compute() local
794 int64_t num_batches = 1; in Compute() local
|
| /aosp_15_r20/external/private-join-and-compute/private_join_and_compute/crypto/ |
| H A D | simultaneous_fixed_bases_exp.cc | 84 size_t num_bases, size_t num_simultaneous, size_t num_batches, in SimultaneousFixedBasesExp()
109 size_t num_batches = (bases.size() + num_simultaneous - 1) / num_simultaneous; in Create() local
126 const Context& context, size_t num_simultaneous, size_t num_batches) { in Precompute()
|
| /aosp_15_r20/packages/modules/NeuralNetworks/runtime/test/specs/V1_2/ |
| D | unidirectional_sequence_rnn.mod.py | 39 def convert_to_time_major(tensor, num_batches, max_time, input_size): argument
44 num_batches = 2 variable
|
| /aosp_15_r20/packages/modules/NeuralNetworks/runtime/test/specs/V1_3/ |
| D | unidirectional_sequence_rnn.mod.py | 42 def convert_to_time_major(tensor, num_batches, max_time, input_size): argument
47 num_batches = 2 variable
|
| /aosp_15_r20/external/ComputeLibrary/tests/validation/reference/ |
| H A D | Winograd.cpp | 280 const int num_batches = in.shape().total_size() / (in_w * in_h * in_d); in winograd_input_transform() local
404 …const int num_batches = in.shape().total_size() / (kernel_size.area() * num_channels * num_filter… in winograd_filter_transform() local
501 const int num_batches = in.shape().total_size() / (w_in * h_in * c_in); in winograd_output_transform() local
|
| H A D | DepthwiseConvolutionLayer.cpp | 68 const int num_batches = src.shape().total_size() / (input_width * input_height * input_depth); in depthwise_convolution_fp() local
161 const int num_batches = src.shape().total_size() / (input_width * input_height * input_depth); in depthwise_convolution_quantized() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/ |
| H A D | LegacyBatchedFallback.cpp | 171 const auto num_batches = c10::multiply_integers(batch_sizes); in batchedTensorInplaceForLoopFallback() local
304 const auto num_batches = c10::multiply_integers(batch_sizes); in batchedTensorForLoopFallback() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/functorch/ |
| H A D | BatchedFallback.cpp | 176 const auto num_batches = c10::multiply_integers(batch_sizes); in batchedTensorInplaceForLoopFallback() local
320 const auto num_batches = c10::multiply_integers(batch_sizes); in batchedTensorForLoopFallback() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/linalg/ |
| H A D | matrix_diag_op.cc | 363 const Eigen::Index num_batches = output.dimension(0); in Compute() local
422 const Eigen::Index num_batches = input.dimension(0); in Compute() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/cl/kernels/ |
| H A D | lstm_full_test.cc | 183 int num_batches() { return n_batch_; } in num_batches() function in tflite::__anone147aac60111::LSTMOpModel
233 const int num_batches = lstm->num_batches(); in VerifyGoldens() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
| H A D | lstm_test.cc | 240 int num_batches() { return n_batch_; } in num_batches() function in tflite::__anon299f347e0111::LSTMOpModel
350 const int num_batches = lstm->num_batches(); in VerifyGoldens() local
2301 int num_batches() { return n_batch_; } in num_batches() function in tflite::__anon299f347e0111::HybridSparseLSTMOpModel
2392 const int num_batches = input.size(); in VerifyGoldens() local
|
| H A D | fully_connected_test.cc | 246 int num_batches() { return batches_; } in num_batches() function in tflite::__anonaa112c3c0111::BaseFullyConnectedOpModel
453 int num_batches() { return batches_; } in num_batches() function in tflite::__anonaa112c3c0111::HybridFullyConnectedOpModel
1598 int num_batches() { return batches_; } in num_batches() function in tflite::__anonaa112c3c0111::SparseFullyConnectedOpModel
|
| /aosp_15_r20/external/rust/android-crates-io/crates/grpcio-sys/grpc/src/core/ext/filters/client_channel/ |
| D | client_channel.cc | 2144 size_t num_batches = 0; in PendingBatchesFail() local
2187 size_t num_batches = 0; in PendingBatchesResume() local
2825 size_t num_batches = 0; in PendingBatchesFail() local
2867 size_t num_batches = 0; in PendingBatchesResume() local
|
| /aosp_15_r20/external/executorch/backends/vulkan/runtime/graph/ops/impl/ |
| H A D | Select.cpp | 70 int32_t num_batches = 1; in add_select_int_node() local
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/distributed/rpc/examples/ |
| H A D | parameter_server_test.py | 25 num_batches = 4 variable
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/batching_util/ |
| H A D | input_split_metadata.cc | 83 const int num_batches = in generate_task_sizes() local
|