Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/tools/graph_transforms/
Dquantize_nodes.cc747 Tensor reshape_dims_tensor(DT_INT32, {1}); in QuantizeNodes() local
748 reshape_dims_tensor.flat<int32>()(0) = -1; in QuantizeNodes()
749 SetNodeTensorAttr<int32>("value", reshape_dims_tensor, &reshape_dims); in QuantizeNodes()