| /external/tensorflow/tensorflow/python/distribute/ |
| D | metrics_v1_test.py | 147 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 …]
|
| /external/tensorflow/tensorflow/core/kernels/ |
| 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()
|
| 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()
|
| 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 …]
|
| 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 …]
|
| D | count_ops.cc | 35 int num_batches = per_batch_counts.size(); in OutputSparse() local
196 int num_batches = is_1d ? 1 : shape.flat<int64>()(0); in Compute() local
270 int num_batches = splits.NumElements() - 1; in Compute() local
|
| 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
|
| D | random_binomial_op.cc | 387 int64 num_batches = 1; in Compute() local
499 int64 num_batches = 1; in Compute() local
|
| 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
|
| D | parameterized_truncated_normal_op.cc | 630 int32 num_batches = shape_tensor.flat<int32>()(0); in Compute() local
792 int64 num_batches = 1; in Compute() local
|
| D | parameterized_truncated_normal_op_gpu.cu.cc | 54 TruncatedNormalKernel(random::PhiloxRandom gen, T* data, int64 num_batches, in TruncatedNormalKernel()
|
| D | range_sampler_test.cc | 274 const int num_batches = 100; in TEST_F() local
|
| /external/tensorflow/tensorflow/python/keras/distribute/ |
| D | keras_metrics_test.py | 122 def _expected_fn(num_batches): argument
|
| /external/tensorflow/tensorflow/core/kernels/linalg/ |
| D | matrix_diag_op.cc | 341 const Eigen::Index num_batches = output.dimension(0); in Compute() local
400 const Eigen::Index num_batches = input.dimension(0); in Compute() local
|
| /external/tensorflow/tensorflow/lite/experimental/kernels/ |
| D | unidirectional_sequence_gru_test.cc | 80 int num_batches() { return n_batch_; } in num_batches() function in tflite::ops::experimental::__anoncf6e42eb0111::GRUOpModel
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/cl/kernels/ |
| D | lstm_full_test.cc | 183 int num_batches() { return n_batch_; } in num_batches() function in tflite::__anonb664ae370111::LSTMOpModel
233 const int num_batches = lstm->num_batches(); in VerifyGoldens() local
|
| /external/libxkbcommon/src/x11/ |
| D | util.c | 178 const size_t num_batches = ROUNDUP(count, SIZE) / SIZE; in adopt_atoms() local
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | fully_connected_test.cc | 201 int num_batches() { return batches_; } in num_batches() function in tflite::__anonae46fc4d0111::BaseFullyConnectedOpModel
345 int num_batches() { return batches_; } in num_batches() function in tflite::__anonae46fc4d0111::HybridFullyConnectedOpModel
1224 int num_batches() { return batches_; } in num_batches() function in tflite::__anonae46fc4d0111::SparseFullyConnectedOpModel
|
| D | basic_rnn_test.cc | 217 int num_batches() { return batches_; } in num_batches() function in tflite::__anon3123f5fe0111::RNNOpModel
|
| D | optional_tensor_test.cc | 192 int num_batches() { return n_batch_; } in num_batches() function in tflite::__anon4dd1434f0111::LSTMOpModel
|
| D | lstm_test.cc | 240 int num_batches() { return n_batch_; } in num_batches() function in tflite::__anonfa79752f0111::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::__anonfa79752f0111::HybridSparseLSTMOpModel
2392 const int num_batches = input.size(); in VerifyGoldens() local
|
| D | unidirectional_sequence_rnn_test.cc | 231 int num_batches() { return batches_; } in num_batches() function in tflite::__anone81b2e4e0111::UnidirectionalRNNOpModel
|
| D | svdf_test.cc | 182 int num_batches() { return batches_; } in num_batches() function in tflite::__anon7126f9820111::BaseSVDFOpModel
|
| D | unidirectional_sequence_lstm_test.cc | 230 int num_batches() { return n_batch_; } in num_batches() function in tflite::__anond8582fe00111::UnidirectionalLSTMOpModel
392 const int num_batches = input.size(); in VerifyGoldens() local
2515 const int num_batches = input.size(); in VerifyGoldens() local
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/gl/ |
| D | api.cc | 106 int num_batches = 0; in Execute() local
|