| /external/tensorflow/tensorflow/lite/kernels/ |
| D | mirror_pad.cc | 88 int Eval(EvalData<T>* eval_data, int current_dim, int flat_index, in Eval() argument
90 if (current_dim == eval_data->num_dims) { in Eval()
92 if (output_index >= eval_data->output_size) { in Eval()
95 eval_data->output_data[output_index] = eval_data->input_data[flat_index]; in Eval()
99 const int cache_index = current_dim * eval_data->input_size + flat_index; in Eval()
100 auto& cache_entry = eval_data->op_data->cache[cache_index]; in Eval()
105 memcpy(eval_data->output_data + output_index, in Eval()
106 eval_data->output_data + cache_entry.first, count * sizeof(T)); in Eval()
111 const int multiplier = (*eval_data->dimension_num_elements)[current_dim]; in Eval()
112 const TfLiteTensor* padding_matrix = eval_data->padding_matrix; in Eval()
[all …]
|
| /external/tensorflow/tensorflow/contrib/eager/python/ |
| D | evaluator.py | 76 return self.call(self._model.eval_data(*args, **kwargs))
210 def call(self, eval_data): argument
362 def call(self, eval_data): argument
364 weights = eval_data.get(self.weights_key, None)
366 self.avg_loss(eval_data[self.loss_key])
367 self.accuracy(eval_data[self.label_key],
368 eval_data[self.predicted_class_key])
370 self.avg_loss(eval_data[self.loss_key], weights=weights)
371 self.accuracy(eval_data[self.label_key],
372 eval_data[self.predicted_class_key],
|
| D | evaluator_test.py | 37 def eval_data(self, d): member in IdentityModel
43 def eval_data(self, d): member in PrefixLModel
53 def call(self, eval_data): argument
54 self.mean(eval_data)
64 def call(self, eval_data): argument
66 self.mean(eval_data["l_outer"])
67 self.sub.call(eval_data["l_inner"])
|
| /external/tensorflow/tensorflow/contrib/model_pruning/examples/cifar10/ |
| D | cifar10_eval.py | 108 eval_data = FLAGS.eval_data == 'test'
109 images, labels = cifar10.inputs(eval_data=eval_data)
|
| D | cifar10_pruning.py | 149 def inputs(eval_data): argument
166 eval_data=eval_data, data_dir=data_dir, batch_size=BATCH_SIZE)
|
| D | cifar10_input.py | 206 def inputs(eval_data, data_dir, batch_size): argument
218 if not eval_data:
|
| /external/tensorflow/tensorflow/contrib/eager/python/examples/rnn_colorbot/ |
| D | rnn_colorbot.py | 213 def test(model, eval_data): argument
216 for (labels, chars, sequence_length) in tfe.Iterator(eval_data):
254 eval_data = load_dataset(
285 test(model, eval_data)
|
| /external/tensorflow/tensorflow/examples/tutorials/layers/ |
| D | cnn_mnist.py | 123 eval_data = mnist.test.images # Returns np.array
150 x={"x": eval_data}, y=eval_labels, num_epochs=1, shuffle=False)
|
| /external/tensorflow/tensorflow/contrib/eager/python/examples/rnn_ptb/ |
| D | rnn_ptb.py | 305 eval_data = _divide_into_batches(corpus.valid, 10)
331 eval_loss = evaluate(model, eval_data)
|
| /external/tensorflow/tensorflow/contrib/autograph/examples/notebooks/ |
| D | dev_summit_2018_demo.ipynb | 1217 "def test(eval_data, lower_cell, upper_cell, relu_layer, batch_size, num_steps):\n",
1219 " iterator = eval_data.make_one_shot_iterator()\n",
1230 …"def train_model(train_data, eval_data, batch_size, lower_cell, upper_cell, relu_layer, train_step…
1234 " test(eval_data, lower_cell, upper_cell, relu_layer, 50, num_steps=tf.constant(2))\n",
1865 " eval_data = load_dataset(data_dir, test_url, 50, training=False)\n",
1875 … " train_data, eval_data, batch_size, lower_cell, upper_cell, relu_layer, train_steps=100)\n",
|