Searched refs:reduction_dims_tensor (Results 1 – 1 of 1) sorted by relevance
757 Tensor reduction_dims_tensor(DT_INT32, {1}); in QuantizeNodes() local758 reduction_dims_tensor.flat<int32>()(0) = 0; in QuantizeNodes()759 SetNodeTensorAttr<int32>("value", reduction_dims_tensor, in QuantizeNodes()