Searched refs:weights_max_id (Results 1 – 2 of 2) sorted by relevance
47 const OpBuilder::TensorID weights_max_id, in AddFullyConnectedHelper() argument70 matmul_op->AddInput(weights_max_id); in AddFullyConnectedHelper()267 const TensorID weights_max_id, in AddFullyConnected() argument271 weights_max_id, graph_builder_, context, in AddFullyConnected()
64 const TensorID weights_max_id,