Searched refs:tensor_values (Results 1 – 5 of 5) sorted by relevance
67 void LogTensorValue(int tensor_index, const float* tensor_values, in LogTensorValue() argument69 tensor_id_to_stats_map_[tensor_index].Update(tensor_values, tensor_size); in LogTensorValue()
88 self.tensor_values = {}97 self.tensor_values[tag] = run_values.results[tag]221 evaled_value = saving_hook.tensor_values[param]249 evaled_value = saving_hook.tensor_values[param]
515 def build_results(self, session, tensor_values): argument536 assert len(self._final_fetches) == len(tensor_values)552 value = tensor_values[j]560 assert j == len(tensor_values)
236 def _log_tensors(self, tensor_values): argument241 logging.info(self._formatter(tensor_values))245 stats.append("%s = %s" % (tag, tensor_values[tag]))
83 typename checkpoint::SaveTypeTraits<T>::RepeatedField* tensor_values = in ValuesFromConstNode() local86 if (!tensor_values->empty() && tensor.has_tensor_shape()) { in ValuesFromConstNode()91 if (shape.dim_size() == 1 && shape.dim(0).size() == tensor_values->size()) { in ValuesFromConstNode()92 values->insert(values->end(), tensor_values->begin(), in ValuesFromConstNode()93 tensor_values->end()); in ValuesFromConstNode()