Home
last modified time | relevance | path

Searched defs:num_batches (Results 1 – 25 of 71) sorted by relevance

123

/external/tensorflow/tensorflow/python/distribute/
Dmetrics_v1_test.py142 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 …]
/external/tensorflow/tensorflow/core/kernels/
Drandom_binomial_op_test.cc27 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()
Dparameterized_truncated_normal_op_test.cc27 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()
Deigen_backward_spatial_convolutions_test.cc231 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 …]
Deigen_pooling_test.cc31 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 …]
Dcount_ops.cc44 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
Deigen_backward_cuboid_convolutions_test.cc299 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
Drandom_binomial_op.cc387 int64_t num_batches = 1; in Compute() local
499 int64_t num_batches = 1; in Compute() local
Deigen_spatial_convolutions_test.cc197 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
/external/private-join-and-compute/private_join_and_compute/crypto/
Dsimultaneous_fixed_bases_exp.cc84 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()
Dpedersen_over_zn.h104 int num_batches = (input.size() + subvector_size - 1) / subvector_size; in SplitVector() local
/external/ComputeLibrary/tests/validation/reference/
DWinograd.cpp280 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
DConvolutionLayer.cpp71 const int num_batches = src.shape().total_size() / (width_in * height_in * depth_in); in convolution_layer_nchw() local
DDepthwiseConvolutionLayer.cpp68 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
DFullyConnectedLayer.cpp141 const int num_batches = dst_shape.total_size_upper(1); in fully_connected_layer() local
/external/executorch/backends/vulkan/runtime/graph/ops/impl/
DSelect.cpp70 int32_t num_batches = 1; in add_select_int_node() local
/external/pytorch/aten/src/ATen/
DLegacyBatchedFallback.cpp171 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
/external/pytorch/torch/testing/_internal/distributed/rpc/examples/
Dparameter_server_test.py25 num_batches = 4 variable
/external/tensorflow/tensorflow/core/kernels/batching_util/
Dinput_split_metadata.cc83 const int num_batches = in generate_task_sizes() local
/external/rust/android-crates-io/crates/ring/benches/
Daead.rs123 let num_batches = (std::cmp::max(1, 8192 / ciphertext.len()) * 10) as u64; localVariable
/external/tensorflow/tensorflow/lite/kernels/
Dunidirectional_sequence_gru_test.cc80 int num_batches() { return n_batch_; } in num_batches() function in tflite::ops::custom::__anon0b9e18fd0111::GRUOpModel
/external/pytorch/aten/src/ATen/functorch/
DBatchedFallback.cpp176 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
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/kernels/
Dlstm_full_test.cc183 int num_batches() { return n_batch_; } in num_batches() function in tflite::__anonf18325460111::LSTMOpModel
233 const int num_batches = lstm->num_batches(); in VerifyGoldens() local
/external/tensorflow/tensorflow/core/kernels/linalg/
Dmatrix_diag_op.cc363 const Eigen::Index num_batches = output.dimension(0); in Compute() local
422 const Eigen::Index num_batches = input.dimension(0); in Compute() local
/external/mesa3d/src/gallium/drivers/iris/i915/
Diris_batch.c173 unsigned num_batches = IRIS_BATCH_COUNT - (devinfo->ver >= 12 ? 0 : 1); in iris_create_engines_context() local

123