Searched refs:key_shape (Results 1 – 1 of 1) sorted by relevance
61 std::initializer_list<uint32_t> key_shape, in HashtableLookupOpModel() argument73 OperandType KeyTy(Type::TENSOR_INT32, key_shape); in HashtableLookupOpModel()