Searched refs:InferOneHotOpType (Results 1 – 1 of 1) sorted by relevance
146 static TensorType InferOneHotOpType(Value indices, Value depth, Value on_value, in InferOneHotOpType() function172 InferOneHotOpType(indices, depth, on_value, off_value, axis), indices, in build()