Home
last modified time | relevance | path

Searched refs:tf_index (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/python/ops/
Dmetrics_impl.py2888 tf_index = math_ops.argmin(math_ops.abs(specificities - specificity), 0)
2889 tf_index = math_ops.cast(tf_index, dtypes.int32)
2892 return math_ops.divide(tp[tf_index],
2893 tp[tf_index] + fn[tf_index] + kepsilon, name)
3730 tf_index = math_ops.argmax(indices_at_minval, 0)
3731 tf_index = math_ops.cast(tf_index, dtypes.int32)
3734 return math_ops.divide(tn[tf_index],
3735 tn[tf_index] + fp[tf_index] + kepsilon, name)
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.cc4929 const int tf_index = tf_pad_index[i]; in ConvertPad() local
4932 pre_padding.d[k] = pad_data[tf_index * 2]; in ConvertPad()
4933 post_padding.d[k] = pad_data[tf_index * 2 + 1]; in ConvertPad()
4934 if (tf_index != trt_index) { in ConvertPad()
4936 std::swap(transpose_idx[tf_index], transpose_idx[trt_index]); in ConvertPad()