/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
|