Constrain randomCategorialLogits to be TensorFlowFloatingPoint in Tensor initializer (#717)
Constrain randomCategorialLogits to be TensorFlowFloatingPoint in Tensor initializer (#717)


tensorflow/swift-apis,tensorflow/swift-apis,tensorflow/swift-apis,tensorflow/swift-apis,tensorflow/swift-apis