| /external/tensorflow/tensorflow/python/distribute/ |
| 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 …]
|
| /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 | 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
|
| 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_t num_batches = 1; in Compute() local
499 int64_t 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 | 637 int32_t num_batches = tensor_shape.dim_size(0); in Compute() local
794 int64_t 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()
|
| /external/ComputeLibrary/tests/validation/reference/ |
| 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
|
| D | ConvolutionLayer.cpp | 71 const int num_batches = src.shape().total_size() / (width_in * height_in * depth_in); in convolution_layer_nchw() local
|
| 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
|
| D | FullyConnectedLayer.cpp | 141 const int num_batches = dst_shape.total_size_upper(1); in fully_connected_layer() local
|
| D | Pooling3dLayer.cpp | 73 const int num_batches = static_cast<int>(src.shape()[idx_batch]); in pooling_3d_layer_internal() local
|
| /external/tensorflow/tensorflow/core/kernels/batching_util/ |
| D | input_split_metadata.cc | 83 const int num_batches = in generate_task_sizes() local
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | unidirectional_sequence_gru_test.cc | 80 int num_batches() { return n_batch_; } in num_batches() function in tflite::ops::custom::__anon253d8dec0111::GRUOpModel
|
| D | unidirectional_sequence_lstm_test_util.h | 228 int num_batches() { return n_batch_; } in num_batches() function
|
| D | optional_tensor_test.cc | 192 int num_batches() { return n_batch_; } in num_batches() function in tflite::__anonbd9bd42d0111::LSTMOpModel
|
| D | basic_rnn_test.cc | 217 int num_batches() { return batches_; } in num_batches() function in tflite::__anon2c82b85c0111::RNNOpModel
|
| /external/tensorflow/tensorflow/core/kernels/linalg/ |
| 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
|
| /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::__anoncb8038550111::LSTMOpModel
233 const int num_batches = lstm->num_batches(); in VerifyGoldens() local
|
| /external/ComputeLibrary/src/runtime/NEON/functions/ |
| D | NELSTMLayer.cpp | 419 const unsigned int num_batches = input->dimension(1); in validate() local
|
| /external/ComputeLibrary/src/runtime/CL/functions/ |
| D | CLLSTMLayer.cpp | 444 const unsigned int num_batches = input->dimension(1); in validate() local
|