Searched refs:weights_t (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | bincount_op.cc | 97 const Tensor& weights_t = ctx->input(2); in Compute() local
105 const auto weights = weights_t.flat<T>(); in Compute()
|
| /external/tensorflow/tensorflow/contrib/metrics/python/ops/ |
| D | metric_ops_test.py | 5951 weights_t = array_ops.placeholder(dtypes_lib.float32, [stride])
5954 predictions_t, labels_t, weights=weights_t)
5962 weights_t: weights[stride * i:stride * (i + 1)]
6082 weights_t = constant_op.constant(
6086 predictions_t, labels_t, weights=weights_t)
6141 weights_t = array_ops.placeholder(dtypes_lib.float32, [stride])
6144 predictions_t, labels_t, weights=weights_t)
6152 weights_t: weights[stride * i:stride * (i + 1)]
6186 weights_t = array_ops.placeholder(dtypes_lib.float32, [stride])
6189 predictions_t, labels_t, weights=weights_t)
[all …]
|