Home
last modified time | relevance | path

Searched refs:metric_fn (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/lite/experimental/quantization_debugger/
Ddebugger.py175 for metric_name, metric_fn in self._layer_debug_metrics.items():
176 layer_statistics[tensor_name][metric_name].append(metric_fn(diffs))
218 metric_fn) in self._debug_options.model_debug_metrics.items():
220 metric_fn(float_tensor_data, quant_tensor_data))
/external/tensorflow/tensorflow/python/keras/engine/
Dtraining_utils_v1.py911 metric_fn = get_metric_function(
915 if not isinstance(metric_fn, metrics_module.Metric):
916 metric_fn = metrics_module.MeanMetricWrapper(
917 metric_fn, name=metric_name)
918 metrics_dict[metric_name] = metric_fn
1124 metric_fn = metrics_module.get(metric)
1126 if hasattr(metric_fn, 'name'):
1127 suffix = metric_fn.name
1129 suffix = metric_fn.__name__
1176 def call_metric_function(metric_fn, argument
[all …]
Dtraining_v1.py1854 for metric_name, metric_fn in metrics_dict.items():
1858 metric_fn._name = metric_name # pylint: disable=protected-access
1859 updated_metrics_dict[metric_name] = metric_fn
1861 self._compile_metric_functions.append(metric_fn)
1912 for metric_name, metric_fn in metrics_dict.items():
1915 metric_fn, y_true, y_pred, weights=weights, mask=mask)
/external/tensorflow/tensorflow/python/tpu/
Dasync_checkpoint_test.py83 def metric_fn(labels, logits): function
92 eval_metrics = (metric_fn, [labels, logits])
/external/tensorflow/tensorflow/python/distribute/
Dmetrics_v1_test.py99 def _test_metric(self, distribution, dataset_fn, metric_fn, expected_fn): argument
106 metric_fn, args=(inputs,))
121 metric_fn, args=(iterator.get_next(),))
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.estimator.pbtxt229 argspec: "args=[\'estimator\', \'metric_fn\'], varargs=None, keywords=None, defaults=None"
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.estimator.pbtxt221 argspec: "args=[\'estimator\', \'metric_fn\'], varargs=None, keywords=None, defaults=None"
/external/tensorflow/tensorflow/python/keras/
Dmetrics_test.py2039 def metric_fn(y_true, y_pred): function
2047 mean_metric = metrics.MeanMetricWrapper(metric_fn)
2048 sum_metric = metrics.SumOverBatchSizeMetricWrapper(metric_fn)