Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/python/distribute/
Dmetrics_v1_test.py147 def _expected_fn(num_batches): argument
160 def _expected_fn(num_batches): argument
176 def _expected_fn(num_batches): argument
195 def _expected_fn(num_batches): argument
213 def _expected_fn(num_batches): argument
232 def _expected_fn(num_batches): argument
246 def _expected_fn(num_batches): argument
259 def _expected_fn(num_batches): argument
272 def _expected_fn(num_batches): argument
285 def _expected_fn(num_batches): argument
[all …]
Dkeras_metrics_test.py124 def _expected_fn(num_batches): argument
/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 …]
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
Dmatrix_diag_op.cc341 const Eigen::Index num_batches = output.dimension(0); in Compute() local
400 const Eigen::Index num_batches = input.dimension(0); 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
Dparameterized_truncated_normal_op_gpu.cu.cc54 TruncatedNormalKernel(random::PhiloxRandom gen, T* data, int64 num_batches, in TruncatedNormalKernel()
Dparameterized_truncated_normal_op.cc339 int32 num_batches = shape_tensor.flat<int32>()(0); in Compute() local
Drandom_binomial_op.cc385 int64 num_batches = 1; in Compute() local
Drange_sampler_test.cc274 const int num_batches = 100; in TEST_F() local
Dnon_max_suppression_op.cc294 const int num_batches = inp_boxes.dim_size(0); in BatchedNonMaxSuppressionOp() local
/external/tensorflow/tensorflow/lite/experimental/kernels/
Dunidirectional_sequence_gru_test.cc80 int num_batches() { return n_batch_; } in num_batches() function in tflite::ops::experimental::__anona14c348f0111::GRUOpModel
/external/libxkbcommon/xkbcommon/src/x11/
Dutil.c162 const size_t num_batches = ROUNDUP(count, SIZE) / SIZE; in adopt_atoms() local
/external/tensorflow/tensorflow/lite/kernels/
Dbasic_rnn_test.cc218 int num_batches() { return batches_; } in num_batches() function in tflite::__anon76108c220111::RNNOpModel
Dfully_connected_test.cc191 int num_batches() { return batches_; } in num_batches() function in tflite::__anon66fc9d710111::BaseFullyConnectedOpModel
328 int num_batches() { return batches_; } in num_batches() function in tflite::__anon66fc9d710111::HybridFullyConnectedOpModel
Doptional_tensor_test.cc194 int num_batches() { return n_batch_; } in num_batches() function in tflite::__anon0686e4730111::LSTMOpModel
Dunidirectional_sequence_lstm_test.cc233 int num_batches() { return n_batch_; } in num_batches() function in tflite::__anon08738b840111::UnidirectionalLSTMOpModel
394 const int num_batches = input.size(); in VerifyGoldens() local
2513 const int num_batches = input.size(); in VerifyGoldens() local
Dunidirectional_sequence_rnn_test.cc230 int num_batches() { return batches_; } in num_batches() function in tflite::__anon46a7a5720111::UnidirectionalRNNOpModel
Dsvdf_test.cc181 int num_batches() { return batches_; } in num_batches() function in tflite::__anon3f7029260111::BaseSVDFOpModel
Dlstm_test.cc237 int num_batches() { return n_batch_; } in num_batches() function in tflite::__anonc8c2a4d30111::LSTMOpModel
361 const int num_batches = input.size(); in VerifyGoldens() local
2312 int num_batches() { return n_batch_; } in num_batches() function in tflite::__anonc8c2a4d30111::LSTMIntegerOpModel
Dbidirectional_sequence_rnn_test.cc789 int num_batches() { return batches_; } in num_batches() function in tflite::__anon28672a310111::BidirectionalRNNOpModel
/external/tensorflow/tensorflow/lite/delegates/gpu/gl/
Dapi.cc106 int num_batches = 0; in Execute() local

12