| /external/tensorflow/tensorflow/core/kernels/ |
| D | sdca_ops_test.cc | 88 const int num_examples) { in SparseExampleIndices() argument
89 const int x_size = num_examples * 4; in SparseExampleIndices()
96 const int num_examples) { in SparseFeatureIndices() argument
97 const int x_size = num_examples * 4; in SparseFeatureIndices()
122 void GetGraphs(const int32 num_examples, const int32 num_sparse_feature_groups, in GetGraphs() argument
179 SparseExampleIndices(g, sparse_features_per_group, num_examples))); in GetGraphs()
184 SparseFeatureIndices(g, sparse_features_per_group, num_examples))); in GetGraphs()
189 NodeBuilder::NodeOut(RandomZeroOrOne(g, num_examples * 4))); in GetGraphs()
196 RandomZeroOrOneMatrix(g, num_examples, dense_features_per_group))); in GetGraphs()
199 Node* const weights = Ones(g, num_examples); in GetGraphs()
[all …]
|
| D | sdca_internal.cc | 247 for (int example_id = 0; example_id < num_examples(); ++example_id) { in SampleAdaptiveProbabilities()
270 auto local_gen = generator.ReserveSamples32(num_examples()); in SampleAdaptiveProbabilities()
284 while (id < num_examples() && num_retries < num_examples()) { in SampleAdaptiveProbabilities()
295 examples_not_seen.reserve(num_examples()); in SampleAdaptiveProbabilities()
296 for (int i = 0; i < num_examples(); ++i) { in SampleAdaptiveProbabilities()
305 for (int i = id; i < num_examples(); ++i) { in SampleAdaptiveProbabilities()
350 const int num_examples = static_cast<int>(example_weights.size()); in Initialize() local
360 examples_.resize(num_examples); in Initialize()
361 probabilities_.resize(num_examples); in Initialize()
362 sampled_index_.resize(num_examples); in Initialize()
[all …]
|
| D | sdca_internal.h | 337 int num_examples() const { return examples_.size(); } in num_examples() function
352 const DeviceBase::CpuWorkerThreads& worker_threads, int num_examples,
363 const DeviceBase::CpuWorkerThreads& worker_threads, int num_examples,
371 const DeviceBase::CpuWorkerThreads& worker_threads, int num_examples,
|
| D | sdca_ops.cc | 145 TensorShape expected_example_state_shape({examples.num_examples(), 4}); in DoCompute()
229 examples.num_examples(), kCostPerUnit, train_step); in DoCompute()
|
| /external/tensorflow/tensorflow/python/debug/examples/v1/ |
| D | debug_keras.py | 35 num_examples = 8
39 xs = np.zeros([num_examples, input_dims])
40 ys = np.zeros([num_examples, output_dims])
42 (xs, ys)).repeat(num_examples).batch(int(num_examples / steps_per_epoch))
|
| /external/tensorflow/tensorflow/tools/gcs_test/python/ |
| D | gcs_smoke.py | 39 def create_examples(num_examples, input_mean): argument
41 ids = np.arange(num_examples).reshape([num_examples, 1])
42 inputs = np.random.randn(num_examples, 1) + input_mean
45 for row in range(num_examples):
208 example_data = create_examples(FLAGS.num_examples, 5)
223 if read_count != FLAGS.num_examples:
226 FLAGS.num_examples))
242 for _ in range(FLAGS.num_examples):
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | parsing_ops.cc | 66 const DimensionHandle& num_examples, in AddRaggedOutputShapes() argument
69 TF_RETURN_IF_ERROR(c->Add(num_examples, 1, &num_splits)); in AddRaggedOutputShapes()
212 DimensionHandle num_examples = c->UnknownDim(); in __anonb04296470502() local
214 num_examples = c->Dim(input, 0); in __anonb04296470502()
222 num_examples, c, &output_idx)); in __anonb04296470502()
291 DimensionHandle num_examples = c->Dim(input, 0); in __anonb04296470702() local
293 c->Matrix(num_examples, c->UnknownDim()); in __anonb04296470702()
374 DimensionHandle num_examples = c->UnknownDim(); in __anonb04296470802() local
376 num_examples = c->Dim(input, 0); in __anonb04296470802()
385 num_examples, c, &output_idx)); in __anonb04296470802()
[all …]
|
| /external/tensorflow/tensorflow/core/util/ |
| D | example_proto_fast_parsing.cc | 2280 bool is_batch, int num_examples, in ParseContextDenseFeatures() argument
2297 dense_shape.AddDim(num_examples); in ParseContextDenseFeatures()
2308 for (int e = 0; e < num_examples; e++) { in ParseContextDenseFeatures()
2343 bool is_batch, int num_examples, in ParseContextSparseFeatures() argument
2369 for (int e = 0; e < num_examples; e++) { in ParseContextSparseFeatures()
2390 out_shape(0) = num_examples; in ParseContextSparseFeatures()
2404 bool is_batch, int num_examples, in ParseContextRaggedFeatures() argument
2417 splits_shape.AddDim(num_examples + 1); in ParseContextRaggedFeatures()
2440 for (int e = 0; e < num_examples; e++) { in ParseContextRaggedFeatures()
2468 if (actual_splits != num_examples + 1) { in ParseContextRaggedFeatures()
[all …]
|
| /external/tensorflow/tensorflow/python/keras/benchmarks/ |
| D | benchmark_util.py | 156 num_examples = x.shape[0]
160 total_num_examples = epochs * num_examples
|
| /external/tensorflow/tensorflow/python/keras/benchmarks/keras_examples_benchmarks/ |
| D | mnist_conv_custom_training_benchmark_test.py | 41 self.num_examples = x_train.shape[0]
229 total_num_examples = epochs * self.num_examples
|
| /external/tensorflow/tensorflow/lite/experimental/examples/lstm/ |
| D | input_data.py | 189 def num_examples(self): member in _DataSet
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/ |
| D | auto_shard_dataset_test.py | 71 num_examples, shuffle): argument
75 for _ in range(num_examples):
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | bidirectional_sequence_rnn_test.cc | 1129 const int num_examples = 64; in TEST() local
1130 for (int k = 0; k < num_examples; k++) { in TEST()
|
| /external/tensorflow/tensorflow/lite/kernels/parse_example/ |
| D | parse_example.cc | 432 int num_examples, int batch_size, in CopyToBuffer() argument
436 for (; i < num_examples; ++i) { in CopyToBuffer()
|