Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/graph/
Dquantize_training.cc199 int tn_size = tensor_names.NumElements(); in ConnectVariablesToSaveOp() local
210 var_nodeouts.reserve(tn_size + var_size); in ConnectVariablesToSaveOp()
217 Tensor new_tensor_names(DT_STRING, TensorShape({tn_size + var_size})); in ConnectVariablesToSaveOp()
218 Tensor new_shape_and_slices(DT_STRING, TensorShape({tn_size + var_size})); in ConnectVariablesToSaveOp()
223 new_tensor_names.flat<string>()(tn_size + i) = var->name(); in ConnectVariablesToSaveOp()
224 new_shape_and_slices.flat<string>()(tn_size + i) = ""; in ConnectVariablesToSaveOp()